nvcc fatal:不支持的 GPU 架构“compute_86”

gol*_*lia 8 ubuntu compiler-errors nvidia nvcc pytorch

我有一个带有此驱动程序的 Nvidia RTX 3090 ti 24GB

CUDA Version: 11.4 

Driver Version: 470.74

18.04.1-Ubuntu SMP

Cuda compilation tools, release 9.1, V9.1.85
Run Code Online (Sandbox Code Playgroud)

我已经找到了这个卡架构,它是Ampere,所以库的版本是compute_86或sm_86(如果我没记错的话)。但是用 nvcc 编译时它给了我返回

nvcc fatal:不支持的 GPU 架构“compute_86”

我运行了 nvcc --help 并且发现了一些奇怪的东西,它返回了我的 gpu-code 和 gpu-architecture

此选项允许的值:'compute_30'、'compute_32'、'compute_35'、'compute_37'、'compute_50'、'compute_52'、'compute_53'、'compute_60'、'compute_61'、'compute_62'、'compute_70'、 'compute_72','sm_30','sm_32','sm_35','sm_37','sm_50','sm_52','sm_53','sm_60','sm_61','sm_62','sm_70','sm_72 '。

所以我缺少任何驱动程序版本或某些必须下载的库,或者我无法使用我的 GPU 进行编译?

ihd*_*hdv 4

在您发布的系统信息中,最后一行

Cuda compilation tools, release 9.1, V9.1.85
Run Code Online (Sandbox Code Playgroud)

表明你的NVCC当前是V9.1(使用时nvcc -V可以确定)。该版本的NVCC太旧,不支持compute_86。出现这种情况的可能原因是您分别安装了 CUDA 工具包(包括 NVCC)和 GPU 驱动程序,且 CUDA 版本不同。您可以按照此官方页面上的说明将其更新到V11.4来解决该问题: developer.nvidia.com/cuda-11-4-2-download-archive。根据我的经验,使用 apt 管理 NVIDIA 驱动程序和 CUDA 工具包常常会导致系统混乱。所以建议使用官方安装程序。如果您之前设置过与 CUDA 相关的环境变量,请记住重置它们以链接到新版本。

要获取其他特定版本的 CUDA,您只需 google“cuda 工具包(版本号)下载”并查找 nvidia 官方网站结果即可。

  • @Raymond这三个环境变量需要重置(在~/.bashrc中):`export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.4/lib64`、`export PATH=$PATH:/usr/local/cuda -11.4/bin`,`导出 CUDA_HOME=/usr/local/cuda-11.4` (3认同)