cuda-nvidia显卡算力

查看算力

https://developer.nvidia.com/cuda-gpus

Geforce RTX 3060       8.0

RTX A5000 sm_86

torch.__version__

torch.cuda.is_available()

torch.cuda.get_arch_list()

nvcc fatal : Unsupported gpu architecture ‘compute_86‘

SourceURL:file:///home/aaa/myword/pytorch.docx

意思是显卡算力太高，安装的cuda版本不支持.

compute_86指显卡计算能力是8.6

解决方法
通过系统环境变量设置降低算力：

export TORCH_CUDA_ARCH_LIST="7.5"解决方法
通过系统环境变量设置降低算力：

export TORCH_CUDA_ARCH_LIST="7.5"

export TORCH_CUDA_ARCH_LIST="8.0"

cuda切换

sudo rm -rf /usr/local/cuda

sudo ln -s /usr/local/cuda-11.0 /usr/local/cuda    #

显卡型号

nvidia-smi -a

查询可支持的最高cuda版本

安装显卡驱动的前提下，终端输入

nvidia-smi

可以看到该电脑可以支持的cuda版本最高是11.4，驱动是向下兼容的，所以cuda版本小于等于11.4的都可以安装上。