ubuntu20.4环境下
如果当前CUDA算力高于当前PyTorch版本支持的CUDA算力:
sm_86代表当前CUDA算力,下一行的sm_xx代表当前PyTorch支持的CUDA算力,比如sm_75。
最后就会爆出这个错误,nvcc fatal,error。
可以更新CUDA和PyTorch,也可以在终端输入
export TORCH_CUDA_ARCH_LIST="7.5"
降低算力,就不会再报这个错误了。
ubuntu20.4环境下
如果当前CUDA算力高于当前PyTorch版本支持的CUDA算力:
sm_86代表当前CUDA算力,下一行的sm_xx代表当前PyTorch支持的CUDA算力,比如sm_75。
最后就会爆出这个错误,nvcc fatal,error。
可以更新CUDA和PyTorch,也可以在终端输入
export TORCH_CUDA_ARCH_LIST="7.5"
降低算力,就不会再报这个错误了。