From 893b889b2f62bff91d9c63931787d553ee19512f Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sat, 25 Apr 2020 14:58:36 +0300 Subject: [PATCH] Cmake is fixed for V100 (CC=7.0) --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index a624b592..0b354dc1 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -65,7 +65,7 @@ else() else() cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCHITECTURES}) message(STATUS "Building with CUDA flags: " "${CUDA_ARCH_FLAGS}") - if (NOT "arch=compute_75,code=sm_75" IN_LIST CUDA_ARCH_FLAGS) + if (NOT "arch=compute_70,code=sm_70" IN_LIST CUDA_ARCH_FLAGS) set(ENABLE_CUDNN_HALF "FALSE" CACHE BOOL "Enable CUDNN Half precision" FORCE) message(STATUS "Your setup does not supports half precision (it requires CC >= 7.5)") endif()