  # 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 52 60 61 70 75")
++set(Caffe_known_gpu_archs "50 52 60 61 70 72 75")
  
  ################################################################################################
  # A function for automatic detection of GPUs installed  (if autodetection is enabled)

-		-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_50,code=sm_50 \
+        -gencode arch=compute_52,code=sm_52 \
+        -gencode arch=compute_60,code=sm_60 \
+        -gencode arch=compute_61,code=sm_61 \
+        -gencode arch=compute_70,code=sm_70 \
+        -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/src/caffe/CMakeLists.txt caffe-1.0.apollo/src/caffe/CMakeLists.txt
--- caffe-1.0.orig/src/caffe/CMakeLists.txt	2017-04-16 00:17:48.000000000 +0800
+++ caffe-1.0.apollo/src/caffe/CMakeLists.txt	2020-05-14 21:42:26.880426465 +0800
@@ -22,7 +22,10 @@
 
 add_library(caffe ${srcs})

diff -aruN caffe-1.0.orig/cmake/Cuda.cmake caffe-1.0/cmake/Cuda.cmake
--- caffe-1.0.orig/cmake/Cuda.cmake	2020-06-05 19:54:00.866888345 +0800
+++ caffe-1.0/cmake/Cuda.cmake	2020-06-05 20:05:36.186847965 +0800
@@ -1,10 +1,6 @@
-if(CPU_ONLY)
-  return()
