• 显卡驱动版本号与CUDA版本号对应关系

1. CUDA 12.3 Update 1 Release Notes — Release Notes 12.3 documentation

  • 查看电脑显卡支持的最高cuda版本:

win+r --> cmd --> nvidia-smi

  • CUDA下载地址:

CUDA Toolkit Archive | NVIDIA Developer

  • cudnn下载地址:

CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

  • pytorch gpu 版本安装与 CUDA 对应关系:

​​​​​​​Previous PyTorch Versions | PyTorch

根据本地安装的cuda版本选择pytorch-gpu版本,若下载速度慢可用清华源,例:pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html -i  https://pypi.tuna.tsinghua.edu.cn/simple 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐