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

来源:互联网 发布:电源仿真软件 编辑:程序博客网 时间:2024/05/29 17:21

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

set(Caffe_known_gpu_archs "30 35 50 52 61")

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

cmake -DCUDA_ARCH_NAME=All ..