本文介绍了如何修改Caffe中的CUDA架构配置。首先需要编辑Caffe目录下的Cuda.cmake文件，更新支持的GPU架构列表。其次，在编译过程中设置正确的编译选项，确保所有可用的GPU架构都能被正确识别。
1.修改caffe目录下cmake文件夹下Cuda.cmake文件，修改下面这一行：
set(Caffe_known_gpu_archs "30 35 50 52 61")2.编译的时候设置编译选项：
cmake -DCUDA_ARCH_NAME=All ..
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
