Windows下简单快速的安装带有GPU的tensorflow

一、安装anaconda

anaconda是一款免费的用于科学计算的 Python 发行版，包含了众多流行的科学计算、数据分析的 Python 包，而且安装很方便，你可以从这里下载 。然而，因为某些原因下载速度会比较慢，当然你可以选择“科学上网”，或者你可以从清华源Tsinghua mirror下载。

---

1. 双击下载好的 Anaconda2-5.2.0-Windows-x86_64.exe文件，出现如下界面，点击 Next 即可。我个人下载的是最新版Anaconda3-5.3.1-Windows-x86_64.exe，因为安装的时候没有存图，所以就从网上参照了别人的图片，但其实每个版本安装过程都一样的。
   
2. 点击 I Agree。
   
3. 选择Just me还是All Users，假如你的电脑有好几个 Users ，才需要考虑这个问题.其实我们电脑一般就一个 User，就我们一个人使用，选择Just me 就好了；如果你的电脑有多个用户，选择All Users，我这里选择了 All User，继续点击 Next 。
   
4. 默认一般是安装到 C:\盘文件夹下。你也可以选择 Browse… ，选择你自己想要安装的文件夹。我这里 C 盘空间充裕，所以我直接就装到默认的地方。点击next。
   
5. 这里我们默认就好，也就是不勾选“添加环境变量”，我们后面自己来添加。点击install。
6. 接下来等待就好了。安装时间看个人电脑配置。
   
7. 最后的话可以选择安装VScode，也可以不安装。
8. 手动添加环境变量。一是anaconda的安装目录；二是anaconda安装目录下scripts文件夹目录；三是anaconda下library下的bin文件夹目录。诸如二图所示：
   
   
   我的三个环境变量的目录分别如下，最好三个都添加。

C:\Anaconda
C:\Anaconda\Scripts
C:\Anaconda\Library\bin

然后，右键“我的电脑”>选择“属性”>高级系统设置>环境变量>Path>编辑>新建，具体如下图所示：

9. 添加完成后可以打开Windows命令行窗口，输入conda --version，如下：
   
   如果出现conda的版本号说明环境变量添加成功。

---

二、创建一个新的环境

1. 为了不污染anaconda的base环境，我强烈推荐大家创建一个新的环境，这也是一个好习惯。打开Anaconda Prompt命令行，输入如下命令即可创建一个名字为tf的新环境，你也可以换一个名字：

conda create -n tf

2. 命令行输入如下命令激活新环境。

conda activate tf

不行的话输入：

activate tf

3. 在安装tensorflow之前，因为上述同样的原因，最好使用如下三行命令把conda的源换成清华源，如果需要知道更多细节的话，请访问Tsinghua Mirror。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes

然后使用如下命令安装自己所需的tensorflow-gpu或者cpu的版本。

conda install tensorflow-gpu=1.10.0  #if you want to install 1.10.0 version
conda install tensorflow  #if you want to install cpu version

当然，非常重要的一个问题是如何选择tensorflow-gpu的版本，因为这决定了你安装下来的cuda版本，如果安装错了就会跟自己电脑的显卡不兼容。

首先，你需要查看自己电脑的显卡型号，以及显卡驱动程序（NVIDIA GPU）的版本；当然，只有显卡是NVIDIA的电脑才可以装GPU的tensorflow，这就不用我多说了。打开Windows命令行(cmd),输入如下命令：

nvidia-smi

这样我们可以看到自己的显卡驱动程序的版本了，我的是398.36。然后根据下图查看自己所需的cuda版本。细节请看 link 。

这样我需要的cuda版本为9.2左右，可以为9.1或者9.0（因为我最后装下来的就是9.0）。然后根据下图选择我们需要的tensorflow-gpu的版本。细节请看link。

于是我们就可以知道自己所需的tensorflow-gpu的版本了。我选择的是tensorflow_gpu-1.11.0。个人觉得只要是匹配cuda 9 的版本都行。于是就可以使用之前说的命令进行安装了。

conda install tensorflow-gpu=1.11.0 

4. 安装过程可能需要一点时间，请耐心等待。安装完成后请把tf新环境的三个目录也添加到系统环境变量里去，最好添加（我没试过不添加有什么问题）。我的是：

C:\Anaconda\envs\tf
C:\Anaconda\envs\tf\Scripts
C:\Anaconda\envs\tf\Library\bin

然后按下图进行测试。

测试完成说明tensorflow-gpu安装完成。
接下来我们可以看一下我安装的cuda版本以及tensorflow-gpu版本：

好的，所有的安装到此就全部大功告成了，希望所有看了这篇文章的人都能有所收获，如果我有讲的不对的地方欢迎批评指正，谢谢！

---

参考：

1. https://blog.csdn.net/u012318074/article/details/77075209/
2. https://www.jianshu.com/p/62f155eb6ac5
3. https://blog.csdn.net/ITLearnHall/article/details/81708148
4. https://wangpei.ink/2019/03/29/Install-TensorFlow-GPU-by-Anaconda(conda-install-tensorflow-gpu)/
5. https://blog.csdn.net/li57681522/article/details/82491617
6. https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
7. https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
8. https://tensorflow.google.cn/install/source_windows