nvcc fatal : Unsupported gpu architecture ‘compute_86‘

洛洛洛洛洛洛洛

于 2021-04-07 20:20:47 发布

阅读量1k

点赞数

分类专栏：程序报错合集

版权

1 篇文章

订阅专栏

使用pytorch 安装使用apex时报错

nvcc fatal : Unsupported gpu architecture ‘compute_86‘
command '/usr/local/cuda/bin/nvcc' failed with exit status 1

主要原因应该是硬件能够支持的算力比较高，能达到8.6，但是cuda11.0支持不了这么高的算力，通过下述脚本，设置环境变量，降低算力要求，即可：

export TORCH_CUDA_ARCH_LIST="7.5"