安装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版本。

安装pytorch_第1张图片

四、确定是否安装成功

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

安装pytorch_第2张图片

安装pytorch_第3张图片

你可能感兴趣的:(pytorch,深度学习,python)