-
显卡驱动版本号与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