一、在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后放会进行安装,可以看图中我用红框框起来的内容决定是否安装。