caffe通过CMakeLists.txt编译适用于自定义GPU ARCH的动态库

1.修改caffe目录下cmake文件夹下Cuda.cmake文件,修改下面这一行:

set(Caffe_known_gpu_archs "30 35 50 52 61")

2.编译的时候设置编译选项:

cmake -DCUDA_ARCH_NAME=All ..

你可能感兴趣的:(笔记,DL)