文章目录
- 安装Anaconda并且创建Pytorch环境
- 下载Anaconda
- 配置环境变量
- anaconda的命令行窗口
- 配置添加新的下载源
- 添加激活新的python环境
- 利用conda安装pytorch
- 添加pytorch镜像源
- 正式安装pytorch
- 查看是否安装成功
- 利用pip安装
安装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都会很快~
冲起来吧!!!