Windows10 anaconda安装GPU版本的pytorch

1.由于pytorch在Windows中不支持Python2,所以决定安装python3版本的pytorch。https://pytorch.org/get-started/locally/

2.首先,由于之前尝试过conda和pip安装pytorch,速度慢,所以决定使用清华的源安装pytorch,anaconda源切换成清华的源之后,我的Python版本是3.7,cuda版本是10.1,所以使用pytorch官方的命令安装,只是最后不需要添加 -c pytorch,不添加表示不适用pytorch的官方通道。

conda install pytorch torchvision cudatoolkit=10.1

 3.输入命令之后开始激动的等待安装,结果过了几秒之后提示我在清华的源中找不到cudatoolkit以及还有其他一些依赖包,重新尝试还是不行

4.这时候开始想是不是Python的版本不对,换一个python3.5的结果还是不行

5.又开始想是不是cuda安装的版本不行,然后cuda版本又切换成10.0,还是不行。

6.其中10.0和10.1的cuda版本均确定安装正常。

7.折腾了几次之后就有点慌了,这时候想是不是清华的源不足,随机添加了中科大的源,以前都是只添加一家的源,这次是清华的源和中科大的源同时添加到anaconda配置中。

8.源都添加到配置中后,在输入上述命令开始安装pytorch,

conda install pytorch torchvision cudatoolkit=10.1

9.紧张的等待之后,咔咔咔,出来了,搜索到安装包了,确认安装,然后开始以光速安装。

10.安装成功之后想测试一下pytorch是否成功:

import torch
print(torch.cuda.is_available())

11.结果输出False,心拔凉的,打开conda list一看,发现是CPU版本的pytorch。刚才只想着依赖的包可以安装成功了,没看到安装到的是CPU版本的pytorch,

12.这时候想清华源和中科大源都添加了,还是没有GPU版本的pytorch,怎么办,使用pytorch官方的源试一下,输入如下命令:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

13.回车之后,发现只需要安装pytorch和torchvision就可以了,新乡,算了就用官方的安装吧,确认安装,发现速度并不是龟速一样缓慢,差不多音速吧,还可以,等待安装完成了。

14.安装完成再测试

import torch
print(torch.cuda.is_available())

15.输出True

16.Game start。

 

 


看其他博客,发现添加清华源之后都可以使用如下命令成功安装pytorch,我试了几次都不好使,难道是电脑原因?不管咋样,安装好了。

conda install pytorch torchvision cudatoolkit=10.1
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值