【Bug解决】nvcc fatal : Unsupported gpu architecture ‘compute_86‘

不吃饭就会放大招

已于 2022-07-02 19:36:36 修改

阅读量2.7w

点赞数 54

分类专栏： # Linux 文章标签： pytorch 深度学习

于 2022-03-08 13:39:08 首次发布

本文链接：https://blog.csdn.net/qq_31347869/article/details/123348901

版权

Linux 专栏收录该内容

16 篇文章

订阅专栏

在Ubuntu 18.04上使用CUDA 11.0、PyTorch 1.10和GeForce RTX 3090时，尝试安装Detectron2时遇到'nvcc fatal: Unsupported GPU architecture 'compute_86''的错误。问题源于CUDA 11.0不支持GPU的8.6算力。通过在.bashrc文件中设置环境变量TORCH_CUDA_ARCH_LIST为'8.0'，然后source更新，可以临时解决此问题。尽管可以考虑升级CUDA到11.6来匹配GPU算力，但目前PyTorch仅支持到CUDA 11.3，使得此方案不可行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错描述

执行 pip install 'git+https://github.com/facebookresearch/detectron2.git' 安装 detectron2 遇到错误：

nvcc fatal   : Unsupported gpu architecture 'compute_86'

在这里插入图片描述
报错环境：

Linux：Ubuntu 18.04
GPU：NVIDIA GeForce RTX 3090
CUDA：11.0
Pytorch：1.10
Python：3.9
GCC：7.5.0

解决方案

错误原因： GPU 3090 算力是 8.6，而当前 CUDA 11.0 不支持算力 8.6。

解决方案： 设置环境变量，降低算力要求。

# 打开配置文件
vi ~/.bashrc
# 在配置文件中添加如下一行
export TORCH_CUDA_ARCH_LIST="8.0"  # 因为是CUDA11.0，对应的算力为8.0
# 保存后source更新
source ~/.bashrc

再次执行报错的安装命令，安装成功：
在这里插入图片描述
解决方案（目前不可行）：将 CUDA 版本升级到支持算力 8.6 即可，但同时需要安装对应的PyTorch。在 CUDA官网可以看到目前最新的 CUDA 版本是 11.6，表示支持算力 8.6 的 GPU。但是上 PyTorch官网却看到目前 PyTorch 支持的 CUDA 最高版本是 11.3。因此这个办法暂时是不可行了，因为 PyTorch 还没有支持。
在这里插入图片描述