cmake modification for higher compute capabilities

fedor-chervinskii · fedor-chervinskii · commit 524a5f7ec4f9 · 2017-01-18T16:54:01.000+03:00
diff --git a/cmake/Cuda.cmake b/cmake/Cuda.cmake
@@ -4,7 +4,7 @@ endif()
 
 # Known NVIDIA GPU achitectures Caffe can be compiled for.
 # This list will be used for CUDA_ARCH_NAME = All option
-set(Caffe_known_gpu_archs "20 21(20) 30 35 50")
+set(Caffe_known_gpu_archs "20 21(20) 30 35 50 52 61")
 
 ################################################################################################
 # A function for automatic detection of GPUs installed  (if autodetection is enabled)
