简单易懂的PyTorch安装教程

步骤一

第一步是在Anaconda下新建名为 “pytorch” 的环境，可以直接在Anaconda软件下通过点击 “Create” 来创建，如下图所示：

【补充】
如果想要在指定的位置安装新建环境，则在管理员模式打开命令行，输入以下命令（其中envs-name为待创建的环境名称，无需手动创建，执行命令后会自动创建该文件夹）：

conda create --prefix=D:\envs-name python=3.8

【附】Linux下安装命令（以安装名为paddle的环境为例）：

conda create -n paddle python=3.8

【拓展】
Linux端如果想要复制一个已经存在的环境A，并重命名为B，则可以使用如下命令：

conda create -n B --clone A

Linux删除某个环境的命令：

conda env remove --name xxx

Windows删除某个环境的命令：

conda remove -n 环境名 --all

Windows端创建环境的命令为：

conda create --prefix=B --clone A

查看目前所有训练环境的命令为

conda env list

步骤二

下载和安装pytorch及torchvision文件（该方法可直接通过代码命令安装cuda）。

首先，打开Anaconda Prompt，在这个命令行窗口中，激活刚才新建的pytorch环境，如下所示：

(base) C:\Windows\system32>activate pytorch

(pytorch) C:\Windows\system32>

安装pytorch的命令如下：

pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp37-cp37m-win_amd64.whl

【注】上面这条命令中的cu92是指适合 CUDA 9.2 版本的 pytorch；
cp37是指适合 python 3.7 版本的 pytorch。

如果在线安装较慢或安装失败，则可以先下载文件到本地（直接点击http://download.pytorch.org/whl/cu92/torch-0.4.1-cp37-cp37m-win_amd64.whl进行下载），而后离线安装，具体离线安装过程如下：

(pytorch) C:\Windows\system32>pip install d:/torch-0.4.1-cp37-cp37m-win_amd64.whl
Processing d:\torch-0.4.1-cp37-cp37m-win_amd64.whl
Installing collected packages: torch
Successfully installed torch-0.4.1

接着，安装torchvision的命令如下：

pip install torchvision==0.4.1

至此，pytorch基本环境安装完毕，测试是否成功安装，如下图所示：

可以看到，如果成功安装，则会输出pytorch的版本号。

安装过程中可能遇到的问题

情况1

报错信息如下：

(pytorch) C:\Windows\system32>pip install d:/torch-0.4.1-cp36-cp36m-win_amd64.whl
ERROR: torch-0.4.1-cp36-cp36m-win_amd64.whl is not a supported wheel on this platform.

可能原因：python 版本不对。
解决办法：查看python的版本号，重新下载并安装正确版本的文件。

References 参考

[1] pytorch安装（python3.7 cuda9.2）