Tensorflow和Pytorch深度学习框架安装教程

目前主流深度学习框架有Tensorflow和pytorch，由于一些原因我只在windows10下安装了以上两个深度学习框架。Tensorflow在16年底就出了在windows下可安装的版本，而pytorch在2018年4月25号也出了可在windows下安装的版本。接下来我将给出最简单的方法来安装深度学习框架。

在安装深度学习框架前，首先需要安装Anaconda3，cuda8.0和cudnn5.1。

1、安装Anaconda3

Anaconda3可在官网上下载，或者去清华镜像网站，地址是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/，我选择的是Anaconda3-4.2.0-Windows-x86_64版本。

下载完成后，我们直接按默认安装即可，这个安装过程比较费时间，大家耐心等待。

2、安装cuda8.0和cudnn5.1

如果你的电脑没有Nvidia显卡的话，请直接跳过第2步。

cuda8.0下载地址是https://developer.nvidia.com/cuda-80-ga2-download-archive。下载完成后直接默认安装即可。

接下来下载cudnn5.1，下载地址是https://developer.nvidia.com/rdp/cudnn-archive。

下载完成后，我们将三个文件分别放到 cuda的相应安装目录下。

3、安装cpu和gpu版本的Tensorflow

如果你的电脑是win10低配电脑，没有Nvidia显卡，那就只能安装CPU版本的Tensorflow了，在cmd中输入：pip install tensorflow==1.0。

如果你的电脑是win10高配电脑，有Nvidia显卡，你可以安装CPU版本的Tensorflow或安装GPU版本的Tensorflow，例如安装GPU版本的Tensorflow，在cmd中输入：pip install tensorflow-gpu==1.0。

测试是否安装好了，如下代码进行测试。

4、安装cpu和gpu版本的Pytorch

如果你的电脑是win10低配电脑，没有Nvidia显卡，那就只能安装CPU版本的Pytorch了，在cmd中输入：pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl。

如果你的电脑是win10高配电脑，有Nvidia显卡，你可以安装CPU版本的Pytorch或安装GPU版本的Pytorch，例如安装GPU版本的Pytorch，在cmd中输入：pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl。

测试是否安装好了，如下代码进行测试。

以上就是最简单方便的在win10，python3.5下来安装Tensorflow和pytorch深度学习框架的方法，大家赶紧去试试吧。