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

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值