安装pytorch

一、在anaconda中创建虚拟环境

打开Anaconda Prompt创建一个虚拟环境。比如要创建一个名字为pytorch的虚拟环境，可以如下输入。其中python=3.7指定该虚拟环境的python版本号。

conda create -n pytorch python=3.7

二、进入新创建的虚拟环境。

创建好虚拟环境后，通过在Anaconda Prompt中输入conda activate pytorch进入刚刚创建的虚拟环境pytorch。

conda activate pytorch

三、安装pytorch

进入pytorch官网（PyTorch），在首页下拉，如下图所示，选择相应的版本，将下图中我用箭头指出的红色方框里的内容输入Anaconda Prompt的命令行中即可开始pytorch的下载与安装。如果自己电脑的显卡支持cuda，则选择相应版本号的cuda，否则选择CPU版本。

 

四、确定是否安装成功

首先输入python进入python环境中。

然后输入import python，如果没有报错说明pytorch安装成功。

接着输入torch.cuda.is_available()，如果返回true，说明这台机器的GPU已经可以被pytorch使用。

 

五、故障说明

1.输入安装pytorch的命令后无反应：即输入安装命令后卡在某一步而不进行pytorch的下载与安装。这种情况建议更换其它版本号的cuda。

2.pytorch安装成功，但torch.cuda.is_available()返回false：这是也建议更换其它的版本。下面两张图是我两次安装pytorch时的情况，其中第一张图最后torch.cuda.is_available()输出的是false，第二张图输出的是true。从两张图中我用红框标出的内容也可以看出第一张图输出false的原因。之所以第二张图能成功是因为我换了版本进行安装。在安装pytorch时会像下图一样提示你将会安装哪些包，在你输入y后放会进行安装，可以看图中我用红框框起来的内容决定是否安装。