nvcc fatal : Unsupported gpu architecture ‘compute_86‘问题解决

设备:显卡3080

环境:Windows10 cuda11.0

语言:python

问题

代码setup时出现nvcc fatal : Unsupported gpu architecture ‘compute_86‘的问题,经查找,发现根本问题是显卡与cuda版本支持的算力不匹配。

nvcc fatal : Unsupported gpu architecture ‘compute_86‘问题解决_第1张图片

可行解决方案

1.升级CUDA版本。由于改代码要求cuda版本为11.0,故没有升级cuda版本。

2.据说30系列显卡都会出现此类问题,暴力解决方法就是换电脑换显卡。

3.本人解决方法,向系统环境变量中添加一项:TORCH_CUDA_ARCH_LIST,值为7.5(也可以改成8.0或8.6试一下,改过之后要重启哦)。注意:由于CUDA与硬件有关,故添加系统环境变量后要重启电脑,重新运行才会通过。

配环境就要配半天,还会踩好多坑,以后有时间会把自己项目中的问题及解决方案跟大家分享。祝好运!

你可能感兴趣的:(python,深度学习)