Anaconda安装pytorch-cuda

1 查看电脑对应cuda版本

1. 【win-R】=> 【cmd】=> 确定
   
2. 执行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

等待时间可能会比较久，不过，值得等候。