Windows10安装Pytorch步骤记录

安装Anaconda并且创建Pytorch环境

下载Anaconda

先从anaconda官网,选择匹配自己系统的版本,安装anaconda;
注意!如果你系统已经装了python的话,不要勾选add anaconda to the system variable选项,这个的意思是自动帮你加到环境变量,但是我们自己加会比较好,要不然有可能会和原有的python冲突
同时可以自行选择是否安装vscode,如果之前已经安装了那就不用,如果没安装但不想用vscode那也别勾选。

配置环境变量

将anaconda的安装目录和其下的scripts目录添加到系统的环境变量中,红圈里是我自己的安装路径,大家需要根据自己安装的情况做更改。
注意是新建
在这里插入图片描述

anaconda的命令行窗口

在这里插入图片描述

配置添加新的下载源

这是因为很多conda依赖包的下载链接是外网,下载很慢,还经常断,用清华镜像会快很多

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

这样就把清华镜像站中anaconda各种依赖包的镜像原添加进了conda命令中
在这里插入图片描述

添加激活新的python环境

还是打开anaconda的prompt命令行窗口,键入

conda create -n pytorch python=3.6

其中pytorch是自行设置的环境名,随便取名就好,python=X.X是在这个环境中安装X.X版本的python,这里用3.6
随后会提示是否安装,键入y即可

安装结束后在anaconda prompt中键入

activate pytorch

这样就激活了这个新环境,此时应该可以看到:
在这里插入图片描述

需要注意的是,该新环境中只有很少的python依赖库,剩下的要看自己需要利用pip install或conda install安装,由于这里我们是配置pytoch的环境,因此没有什么额外需要的,就不管它,接下来介绍一些对环境的操作:

激活环境已经介绍过了
安装包:conda install -n your_env_name xxx;
或者激活环境后利用pip install xxx
退出环境:deactivate your_env_name
删除环境:conda remove -n your_env_name --all
查看所有环境:conda env list

至此,pytorch环境创建成功

利用conda安装pytorch

添加pytorch镜像源

将清华镜像源添加到conda install的搜索channels中
打开anaconda prompt命令行窗口,键入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

正式安装pytorch

进入pytorch官网,选择适合自己电脑系统的版本,在这里查看不同显卡驱动所对应的cuda版本,我电脑没有GPU,就只能选NONE了
在这里插入图片描述

conda install pytorch torchvision torchaudio cpuonly

收集好下载包,按y进入安装即可。

然而,我就是这里遇到了坑,我的显卡驱动版本对应9.2的cuda,但是报错Solving environment: failed with initial frozen solve. Retrying with flexible solve,之后显示package not found error,如果各位伙伴也出现相同问题,那别再纠结conda安装了,不管怎样离线安装也好、搜索包也好,都是不行的,但是用pip可以顺利解决;当然,如果没报错顺利安装那是最好。

查看是否安装成功

在pytoch环境下在prompt中键入

import torch
import torchvision
print(torch)
#结果为<module 'torch' from 'xxx\\torch\\__init__.py'>
#xxx为安装的路径

如果没有报错,则代表安装成功,开始进击吧!但推荐看一下下一个模块,有一个很方便的pip镜像源

利用pip安装

将清华镜像源添加到pip的搜索目录中,打开anaconda prompt,键入

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

这是清华镜像中python依赖库的部分,基本涵盖了所有库,而且下载超快,键入这行命令后就会从中下载之后pip install的库

进入pytorch官网,选择适合自己电脑系统的版本,在这里查看不同显卡驱动所对应的cuda版本
在这里插入图片描述

打开anaconda prompt,激活pytorch环境

activate pytorch

再键入该command,同时在command后面加上-i https://pypi.tuna.tsinghua.edu.cn/simple,即键入

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

按y确认安装即可

不出所料,我又在这里遇到了坑,不管我怎样配置,依然在从官网下载安装包,并且下载速度仅有几十kb/s,遇到这种情况的伙伴别急,继续往下看,很轻松就可以解决,没有遇到的真的羡慕你萌(哭)
解决方法:
1.找到这个command里涉及的官网下载链接
2.迅雷下载!
3.为了一致性,我将其下载到了pytorch环境下的site-packages中,我的路径是C:\apps\Anaconda\envs\pytorch\Lib\site-packages
4.打开anaconda prompt
5.激活pytoch环境
6.键入

pip install C:\apps\Anaconda\envs\pytorch\Lib\site-packages\torch-1.4.0+cu92-cp37-cp37m-win_amd64.whl

其中路径已经说过,最后的包名称是你下载的那个,注意要加.whl
7。键入y确认安装
8.再次键入

pip install torchvision	

这是安装torchvision及pytoch的各种依赖库,否则import torch依然会报错

验证是否安装成功:
在pytoch环境下在prompt中键入

import torch
import torchvision
print(torch)
#结果为<module 'torch' from 'xxx\\torch\\__init__.py'>
#xxx为安装的路径

至此,安装完成,而且以后的pip install都会很快~
冲起来吧!!!

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值