caffe 编译 + nvcc fatal : Unsupported gpu architecture ‘compute_86‘

在使用CUDA 8.0编译Caffe (GPU版) 时,遇到nvcc fatal错误:Unsupported gpu architecture ‘compute_86‘。该问题发生在win11环境下,配备NVIDIA RTX3080显卡。解决方法是修改Cuda.cmake文件,将"Fermi"更改为"Ampere",并添加对应RTX3080的GPU架构设置。此外,还需更新build_win.cmd和WindowsDownloadPrebuiltDependencies.cmake配置,参考相关博客链接,完成修改后编译成功。
摘要由CSDN通过智能技术生成

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") 行)</

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值