1.安装说明
本教程使用Anaconda建立Pytorch虚拟环境来安装Pytorch。
2.为什么要使用Anaconda虚拟环境安装Pytorch?
因为环境中通常需要安装很多软件,例如:我同时在使用tensorflow框架。但是他们所需要的Python的关联模块或版本会有所差异。如果都装在一个环境中难免会引起冲突。所以,选择虚拟环境能很好地避免环境之间的冲突。
首先,在anaconda新建虚拟环境,然后在虚拟环境中安装pytorch。
具体操作步骤如下:
(1)打开Anaconda Prompt
(2)建立Pytorch Anaconda虚拟环境
conda create --name pytorch python=3.7 anaconda
其中,python=3.7意味着安装3.7版本的python,当然也可以选择其他版本。加入最后一项anaconda命令建立虚拟环境时,会同时安装例如Notebook、Numpy、Scipy、Matplotlib、Pandas等数据分析包,用户也可以不加,选择后续安装。
(3)启动Pytorch Anaconda虚拟环境
activate pytorch
(4)安装Pytorch
conda install pytorch torchvision cpuonly -c pytorch
详情参见pytorch官网,具体命令根据你的环境选择。
可是我在下载的时候遇到一些问题,那就是下载不动,如下图:
在网上搜了许多解决方法,终于成功,解决方法传送门:成功解决
(6)测试
命令行输入python
进入python,并输入下面代码:
import torch
import torchvision
print(torch.__version__)
成功打印,测试成功,使用exit()
退出python。
在退出python之后,在界面输入以下代码:
conda install -c https://conda.binstar.org/menpo opencv
如果需要使用此虚拟环境在Pycharm进行项目开发,将设置里的Project Interpreter改为相应Anaconda文件目录下的Pytorch虚拟环境中的python.exe文件即可。