WIn10+Anaconda环境下安装PyTorch(避坑指南)

这些天安装 pytorch，遇到了一些坑，特此总结一下，以免忘记。分享给大家。

首先，安装环境是：操作系统 win10，已经预先暗转了 anaconda。

1. 为 pytorch 创建虚拟环境

关于 anaconda 的安装步骤这里就忽略不讲了，win10 下安装 anaconda 非常简单。

安装 anaconda 完毕后，我们在安装 pytorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中，我们通常会根据自己的需求去下载各种相应的框架库，但是可能每个项目使用的框架库并不一样，或使用框架的版本不一样，这样需要我们根据需求不断的更新或卸载相应的库，管理起来相当麻烦。所以通过创建虚拟环境，相当于为不同的项目创建一块独立的空间，在这个空间里，你安装任何库和框架都是独立的，不会影响到外部环境。

总之就是想象成为 pytorch 单独创建一个空间，这个空间与其它空间互不影响。我之前安装了 tensorflow 了，也是创建了一个虚拟环境，且 python 版本是 3.5 的。所以，这里当然也要创建虚拟环境了。

创建 pytorch 虚拟环境步骤：

1.打开 anaconda 自带的 anaconda prompt：

2.打开 anaconda prompt 之后，在命令行输入：

conda create --name pytorch python=3.6

注意，这里的 pytorch 是虚拟环境的名称，可随意取。

3.然后继续下列命令，进入 pytorch 虚拟环境：

activate pytorch

好了，pytorch 的虚拟环境创建好了。接下来就是安装 pytorch 了。

其实，安装 pytorch 还是非常简单的。接下来记录一些坑。

2. pytorch 官网安装

pytorch 的官网提供了简单的安装方法，只需简单的命令即可。

首先，打开 pytorch 官网安装页面（需自备*）：

可以看到页面里的这张图，非常简单。我们的环境是：windows、conda、python 3.6、none cuda，对应的命令是：

conda install pytorch-cpu torchvision-cpu -c pytorch

复制这条命令在 anaconda prompt 里输入，则 pytorch “顺利”安装完成。

但是问题来了，在 jupyter 里 import torch 没有问题，可 import torchvision 确报错了：

importerror: dll load failed: 找不到指定的模块。

可能是 pillow 模块有问题。尝试卸载 pillow 模块，但是会同时卸载 pytorch，故为一坑。

3. pip 安装

上一种方法舍弃，这里尝试使用 pip 安装。

同样在 anaconda prompt 里输入以下命令：

# python 3.6
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision

安装顺利完成。

这时候在 jupyter 里 import torch 没有问题，可 import torchvision 确仍然报错了：

importerror: dll load failed: 找不到指定的模块。

解决方案：

在 anaconda prompt 里输入以下命令：

conda uninstall pillow
pip install pillow

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

至此，填坑。pytorch 成功安装。以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。