1 查看电脑对应cuda版本
- 【win-R】=> 【cmd】=> 确定
- 执行
nvidia-smi
,可以从图中看出,该电脑的CUDA Version为:11.6
2 官网查找对应的版本的安装语句
PyTorch官网:https://pytorch.org/
2.1 可以直观的选择对应的版本
如果没有在以下版本中,请点击下方的Previous versions of PyTorch
2.2 Previous versions of PyTorch
按下图的说明,选取合适的pytorch版本
3 Anaconda中使用命令行安装pytorch
Anaconda Prompt 中,打开对应的环境(如下图,打开了forall),安装pytorch
# 打开forall环境
conda activate forall
# 执行命令,安装pytorch,torchvision,torchaudio,pytorch-cuda
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
等待时间可能会比较久,不过,值得等候。