前言

本篇文章讲述如何利用anaconda安装pytorch环境,以及在jupyter中调用对应环境。

系统环境以及所需工具

  1. Windows 10
  2. anaconda(官方下载链接
选择自己所需要的版本,这里我选择的是python3.7 下 64-Bit 这个版本,安装过程不做解释,这个过程实属简单。

安装过程

注:安装之前建议更换conda源,原镜像实在是太慢,我改的是清华源,具体细节可以百度

1. 打开cmd命令行
2. 安装ipykernel(这是为了方便之后用jupyter加载当前所要创建的pytorch环境所需要安装的包,创建环境的时候用)
conda install ipykernel
3. 并输入配置命令(这一步也可以用图形化界面完成,但考虑到linux版本anaconda没有图形化界面,建议还是通过命令行完成)
conda create -n pytorch python=3.7 ipykernel

#以下是命令各个参数的说明
pytorch					环境名称
python=3.7				指定的python版本
ipykernel				第一步所安装的包
4. 激活环境
conda activate pytorch

#当前面出现pytorch,就算激活了

Aanaconda环境下配置pytorch时报错_python

5. 接下来登陆pytorch官网(pytorch官方下载链接

根据自己机器选择对应环境,复制命令到命令行。

Aanaconda环境下配置pytorch时报错_python_02


如果推荐中没有与本机所装cuda一致的选项,那就选择左下角的小字,或者回到本文章点下方链接也可,找到符合本机环境的版本并复制对应命令。

previous versions of pytorch

6. 安装pytorch

在所创建的环境下将复制好的命令输入命令行,官方命令后面有个 -c pytorch,换过源的一定要把这个删掉再输入,因为如果有这个会直接连接官方下载,会非常慢,删掉后才会连接所更换的源。

Aanaconda环境下配置pytorch时报错_pytorch_03

7.静静等

等待安装,需要下不少东西,不一定能一次成功,中间有可能断联,断了以后重新下就ok。

8.最后需要解决一个问题

以上安装的环境下,import torch没问题,import torchvision会报错,这是pillow版本问题,解决方法也很简单。

进入conda中pytorch环境

pip uninstall pillow

pip install pillow==6.1