简单易懂的PyTorch安装教程

步骤一

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

步骤二

下载和安装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）