caffe (GPU版) 编译出现错误:
nvcc fatal : Unsupported gpu architecture ‘compute_86‘
(出现以上错误消息时的caffe环境配置:
1. win11 + NVIDIA RTX3080
2. cmake-3.16.8-win64-x64.msi
3. en_visual_studio_community_2015_with_update_3_x86_x64_dvd_8923300.iso
4. Anaconda3-2021.05-Windows-x86_64.exe
5. cuda_8.0.61_win10.exe; cudnn-8.0-win-x64-v5.1.zip.zip )(标注顺序即为安装顺序)
解决办法:
打开caffe-windows文件夹中的文件Cuda.cmake,路径:.\caffe-windows\cmake\Cuda.cmake,修改如下图所示中的内容。
即:将63行红色方框中的原本的 "Fermi",改为如图所示的内容(Ampere为显卡 RTX3080 的 GPU 架构)。在第78行的下面,添加紧贴此行的红色方框中的内容,在100行的下面,修改为如图中最后一个红色方框的内容( 注释第101行,并在第101行的下面添加 set(__cuda_arch_bin "60 61") 行)</