NVIDIA A30使用PyTorch训练模型时报错:NVIDIA A30 with CUDA capability sm_ 80 is not compatible with the current PyTorch installation
出现问题的原因
- 正常来说,不是向下兼容安装的pytorch的话,与pytorch版本无关
- a30显卡算力是80,目前仅cuda11+支持
解决办法
- 我现有的cuda10.2, 再装个cuda 11.2
- 然后修改~/.bashrc, 指向cuda11.2的地址,弃用10.2
# cuda
export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/