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