updated cmake nvcc flags to work with Tesla V100

cpheinrich · Mar 22, 2019 · 4a58c89 · 4a58c89
1 parent bf82758
commit 4a58c89
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -2,12 +2,12 @@ cmake_minimum_required (VERSION 2.8)
 project (fusibile)
 
 find_package(OpenCV REQUIRED )
-find_package(CUDA 6.0 REQUIRED ) # For Cuda Managed Memory and c++11
+find_package(CUDA 9.0 REQUIRED ) # For Cuda Managed Memory and c++11
 
 include_directories(${OpenCV_INCLUDE_DIRS})
 include_directories(.)
 
-set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS};-O3 --use_fast_math --ptxas-options=-v -std=c++11 --compiler-options -Wall -gencode arch=compute_60,code=sm_60 -gencode arch=compute_60,code=sm_60)
+set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS};-O3 --use_fast_math --ptxas-options=-v -std=c++11 --compiler-options -Wall -gencode arch=compute_70,code=sm_70 -gencode arch=compute_70,code=sm_70)
 
 if(CMAKE_COMPILER_IS_GNUCXX)
     add_definitions(-std=c++11)