 +                                  VERSION_VAR   CUDNN_VERSION)
 +
 +
-diff -aruN caffe-1.0.orig/CMakeLists.txt caffe-1.0.apollo/CMakeLists.txt
---- caffe-1.0.orig/CMakeLists.txt	2017-04-16 00:17:48.000000000 +0800
-+++ caffe-1.0.apollo/CMakeLists.txt	2020-05-14 21:50:43.095191423 +0800
+diff -aruN caffe-1.0.orig/CMakeLists.txt caffe-1.0/CMakeLists.txt
+--- caffe-1.0.orig/CMakeLists.txt	2020-06-05 19:54:00.866888345 +0800
++++ caffe-1.0/CMakeLists.txt	2020-06-05 19:54:09.626775454 +0800
 @@ -8,6 +8,8 @@
  
  # ---[ Caffe project

   if(OSX_VERSION VERSION_GREATER 10.8)
     # enabled by default if and only if CUDA version is less than 7.0
     caffe_option(USE_libstdcpp "Use libstdc++ instead of libc++" (CUDA_VERSION VERSION_LESS 7.0))
diff -aruN caffe-1.0.orig/cmake/Dependencies.cmake caffe-1.0.apollo/cmake/Dependencies.cmake
--- caffe-1.0.orig/cmake/Dependencies.cmake	2017-04-16 00:17:48.000000000 +0800
+++ caffe-1.0.apollo/cmake/Dependencies.cmake	2020-05-14 21:35:45.916540790 +0800
@@ -5,7 +5,7 @@
 set(Caffe_COMPILE_OPTIONS "")
 

-		-gencode arch=compute_60,code=sm_60 \
-		-gencode arch=compute_61,code=sm_61 \
-		-gencode arch=compute_61,code=compute_61
+CUDA_ARCH :=    -gencode arch=compute_60,code=sm_60 \
+        -gencode arch=compute_61,code=sm_61 \
+        -gencode arch=compute_70,code=sm_70 \
+        -gencode arch=compute_72,code=sm_72 \
+        -gencode arch=compute_75,code=sm_75 \
+        -gencode arch=compute_75,code=compute_75
 
 # BLAS choice:
 # atlas for ATLAS (default)
diff -aruN caffe-1.0.orig/scripts/travis/configure-cmake.sh caffe-1.0/scripts/travis/configure-cmake.sh
--- caffe-1.0.orig/scripts/travis/configure-cmake.sh	2020-06-05 19:54:00.902887880 +0800
+++ caffe-1.0/scripts/travis/configure-cmake.sh	2020-06-05 20:08:54.764749042 +0800
@@ -16,8 +16,8 @@
 fi
 
 if $WITH_CUDA ; then
-  # Only build SM50
-  ARGS="$ARGS -DCPU_ONLY=Off -DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN=\"50\" -DCUDA_ARCH_PTX=\"\""
+  # Only build SM61
+  ARGS="$ARGS -DCPU_ONLY=Off -DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN=\"61\" -DCUDA_ARCH_PTX=\"\""
 else
   ARGS="$ARGS -DCPU_ONLY=On"
 fi
diff -aruN caffe-1.0.orig/scripts/travis/configure-make.sh caffe-1.0/scripts/travis/configure-make.sh
--- caffe-1.0.orig/scripts/travis/configure-make.sh	2020-06-05 19:54:00.902887880 +0800
+++ caffe-1.0/scripts/travis/configure-make.sh	2020-06-05 20:07:49.521433379 +0800
@@ -24,8 +24,8 @@
 fi
 
 if $WITH_CUDA ; then
-  # Only build SM50
-  LINE "CUDA_ARCH := -gencode arch=compute_50,code=sm_50"
+  # Only build SM61
+  LINE "CUDA_ARCH := -gencode arch=compute_61,code=sm_61"
 else
   LINE "CPU_ONLY := 1"
 fi
diff -aruN caffe-1.0.orig/src/caffe/CMakeLists.txt caffe-1.0/src/caffe/CMakeLists.txt
--- caffe-1.0.orig/src/caffe/CMakeLists.txt	2020-06-05 19:54:00.902887880 +0800
+++ caffe-1.0/src/caffe/CMakeLists.txt	2020-06-05 19:54:09.626775454 +0800
@@ -22,7 +22,10 @@
 
 add_library(caffe ${srcs})
