error `nvcc fatal : Unsupported gpu architecture ‘compute_80‘ ninja: build stopped: subcommand fai

本文介绍了解决在安装代码setup.py过程中遇到的“Unsupported gpu architecture”错误的方法。此错误源于代码中指定的GPU架构高于本机GPU的实际算力。文章提供了具体的修改建议,并附带了相关代码。

问题:

Error/问题

nvcc fatal   : Unsupported gpu architecture 'compute_80'
ninja: build stopped: subcommand failed.

解决方案

在安装一个代码 setup.py (见下面的代码)里的时,会出现以上报错. 这里 是因为,代码与gpu算力不匹配的原因。 对于本机2080ti, 其最高算力为7.5, 故不支持compute_80(算力8.0), compute_86(算力8.6),解决办法就是把对应两行注释掉就可以了。
关于,gpu算力,可以参考:https://blog.youkuaiyun.com/chengxy1998/article/details/121924208?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2aggregatepagefirst_rank_ecpm_v1~rank_v31_ecpm-2-121924208.pc_agg_new_rank&utm_term=GPU%E7%AE%97%E5%8A%9B%E9%97%AE%E9%A2%98&spm=1000.2123.3001.4430

setup(
    name='droid_backends',
    ext_modules=[<
### 编译 DCNv2 时遇到 `nvcc` 不支持 GPU 架构 `'compute_86'` 的解决方案 当尝试编译带有特定 CUDA 版本的项目如 DCNv2 并遭遇 `nvcc fatal : Unsupported gpu architecture 'compute_86'` 错误时,这通常意味着当前使用的 NVIDIA 编译器工具链版本不兼容所指定的目标架构。 #### 修改 Makefile 或 CMakeLists.txt 文件 对于此类问题的一个常见修复措施是在项目的构建配置文件中调整目标计算能力设置。具体来说,在 `Makefile` 或者 `CMakeLists.txt` 中找到定义 CUDA 架构的地方,并将其更改为与本地环境相匹配的支持版本: ```makefile CUDA_ARCH := -gencode arch=compute_35,code=sm_35 \ -gencode arch=compute_50,code=sm_50 \ -gencode arch=compute_60,code=sm_60 \ -gencode arch=compute_70,code=sm_70 \ -gencode arch=compute_75,code=sm_75 \ -gencode arch=compute_80,code=sm_80 \ -gencode arch=compute_86,code=sm_86 # 移除或注释掉这一行 ``` 如果使用的是较新的 CUDA 工具包,则可能需要移除或修改上述代码片段中的最后一行,因为并非所有的 CUDA 发布版都支持最新的 GPU 架构[^1]。 #### 更新 CUDA 安装路径 另一个潜在的原因可能是由于系统上的 CUDA 路径配置不当引起的。确认 `.bashrc` 和其他 shell 配置文件里关于 CUDA 的环境变量指向了正确的安装位置。例如: ```shell export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 确保这些路径确实对应于已安装的 CUDA 版本,并且该版本能够识别并处理所需的目标架构[^4]。 #### 使用合适的 PyTorch 及其依赖库版本 考虑到不同版本之间的兼容性差异,建议检查正在使用的 Python、PyTorch 和 CUDA 组合是否相互兼容。特别是对于深度学习框架而言,某些预编译二进制文件可能会针对特定硬件特性进行优化,因此选择适当组合非常重要[^5]。 通过以上方法应该可以有效解决因 `nvcc` 不支持某一代号为 `compute_86` 的 GPU 架构而导致的编译失败问题。
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值