1.修改caffe目录下cmake文件夹下Cuda.cmake文件,修改下面这一行:
set(Caffe_known_gpu_archs "30 35 50 52 61")
2.编译的时候设置编译选项:
cmake -DCUDA_ARCH_NAME=All ..
1.修改caffe目录下cmake文件夹下Cuda.cmake文件,修改下面这一行:
set(Caffe_known_gpu_archs "30 35 50 52 61")
2.编译的时候设置编译选项:
cmake -DCUDA_ARCH_NAME=All ..