GPU算力太高导致nvcc fatal : Unsupported gpu architecture ‘compute_86‘

ubuntu20.4环境下

如果当前CUDA算力高于当前PyTorch版本支持的CUDA算力:
GPU算力太高导致nvcc fatal : Unsupported gpu architecture ‘compute_86‘_第1张图片
sm_86代表当前CUDA算力,下一行的sm_xx代表当前PyTorch支持的CUDA算力,比如sm_75。
GPU算力太高导致nvcc fatal : Unsupported gpu architecture ‘compute_86‘_第2张图片
最后就会爆出这个错误,nvcc fatal,error。

可以更新CUDA和PyTorch,也可以在终端输入

export TORCH_CUDA_ARCH_LIST="7.5"

降低算力,就不会再报这个错误了。

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