Skip to content

Instantly share code, notes, and snippets.

@andreibosco

andreibosco/CUDA-compute-fix.patch

Last active July 30, 2021 16:24
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save andreibosco/2a896da79b8405f2cd016acc6aa71a87 to your computer and use it in GitHub Desktop.
Save andreibosco/2a896da79b8405f2cd016acc6aa71a87 to your computer and use it in GitHub Desktop.
Download ZIP
Caffee CUDA compute fix
Raw
CUDA-compute-fix.patch
--- cmake/Cuda.cmake 2021-07-30 12:07:44.707798734 -0400
+++ cmake/Cuda.cmake.fixed 2021-07-30 12:22:30.551329548 -0400
@@ -4,7 +4,7 @@
# 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 60 61")
+set(Caffe_known_gpu_archs "30 35 50 60 61")
################################################################################################
# A function for automatic detection of GPUs installed (if autodetection is enabled)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment