#终端里输入
nvidia-smi
算力查询：CUDA GPU 计算能力 | NVIDIA 开发者
显卡支持的cuda版本：CUDA Toolkit, Driver, and Architecture Matrix — NVIDIA Data Center Drivers
Ongoing表示最新的cuda版本一直支持。
由以上可知，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()}')"
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。