新手安装pytorch

下载anaconda

官网下载

生成虚拟环境

conda create --name pytorch python=3.6

pytorch为虚拟环境的名字
虚拟环境的其他操作Anaconda创建虚拟环境及更换pip安装源

安装成功后，输入以下指令：

conda info --envs
即可看到 conda 环境中，有新建的 pytorch 环境，右边的 * 号表示，当前你处于哪个环境。

(base) D:\>conda info --envs
# conda environments:
#
base                  *  D:\Anaconda3
python3                  D:\Anaconda3\envs\python3

切换到虚拟环境

接下来，我们要在 pytorch 环境中，安装 PyTorch，（有点绕），使用如下指令，进入 pytorch 环境。

conda activate pytorch

你可以看到左边的 base 变成了 pytorch，代表成功进入 pytorch 环境。

安装cpu的pytorch

如果你没有gpu环境，想使用Pytorch，打开Pytorch的官网：https://pytorch.org/

在首页我们可以看到有各种配置可选，我们这里选择CPU的版本（CUDA部分选择None）：

安装demo为：

conda install pytorch torchvision cpuonly -c pytorch

或者

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

import torch报错

如果错误是
No module named ‘torch’

遇到问题，就要解决问题，方法如下：
打开 Anaconda Navigator;
切换到安装 PyTorch 虚拟环境 （如下图）：

3. 安装 Jupyter Notebook（如下图）；

4. 重启 Joupyter Notebook，即可

import torchvision 报错

如果安装 在 Jupyter 里 import torch 没有问题，可 import torchvision 确报错了：

ImportError: DLL load failed: 找不到指定的模块。

那就在 Anaconda Prompt 里输入以下命令：

conda uninstall pillow
pip install pillow

使用 pip 重新安装 pillow 模块即可。