显卡、cuda、pytorch版本确定与安装

1、你的显卡和算力

#终端里输入
nvidia-smi

算力查询：CUDA GPU 计算能力 | NVIDIA 开发者

 2、cuda版本

显卡支持的cuda版本：CUDA Toolkit, Driver, and Architecture Matrix — NVIDIA Data Center Drivers

Ongoing表示最新的cuda版本一直支持。

3、pytorch安装

由以上可知，cuda≥12.8

稳妥起见，不选择最新，见下图。pytorch地址：Get Started

 python≥3.10，然后命令行安装pytorch即可，对应的cuda、cudnn等会自动安装上。

安装完毕后，检查下

python -c "import torch; print(f'PyTorch版本: {torch.__version__}'); print(f'CUDA可用: {torch.cuda.is_available()}'); print(f'支持的算力: {torch.cuda.get_arch_list()}')"