PyTorch安装教程

由于学业上会用到PyTorch，所以今天花了一些时间来安装，实际也踩了一些坑。

我原本已经下载好了anaconda，实际下载也很简单，故不过多赘述。

我自己的python版本是3.9

1.进入anaconda prompt

输入

nvidia-smi

就会呈现出你cuda的版本号，本人是11.7

这里要注意，我原本的版本号是11.0，但是pytorch官网上适合我的版本只有10.2（如下图），（须知 所选版本必须要低于自己的版本）所以我建议可以去官网下载一下最新版的cuda，我下载了11.7的，因此在pytorch官网上可以选择cuda11.6

 

 2.创建虚拟环境

在anaconda prompt上输入

conda create -n PyTorch python=3.9

PyTorch是我定义的环境名称

我的python版本是3.9

故遵从格式：

conda create -n xxxxx（自定义的虚拟环境名称） python=3.x （你的python版本）

 按y进行下一步

创建成功后，可以输入：

conda info --envs

来检查是否创建成功，如下图

 3.进入虚拟环境

conda activate PyTorch

4.复制pytorch官网的代码，进行安装

比如我的是

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

一开始我cuda为11.0时，遇到的问题是

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

后来我如1.所讲，更新了cuda到11.7，就可以安装了

然后就是慢慢等待和按y，由于网速还可以，就没有换清华源

5.下载好后测，显示ture，大功告成