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()