Windows10下安装TensorFlow 1.5-GPU版

[hiyouga]

搞了好几个小时终于配置好了TensorFlow，激动地泪流满面，赶紧写一篇blog记录一下安装过程_(:з」∠) _

环境配置：
Windows 10 版本1709 基于x64处理器
NVIDIA GTX 960M
Python 3.6.3

• 首先通过pip安装TensorFlow-gpu：

pip3 install --upgrade tensorflow-gpu

我还听着网上的教程安装了Visual Studio 2017，但我个人感觉作用不大，更新一下Visual C++ Redistributable应该就可以了，所以这步是否需要还有待考证
（实测需要Microsoft Visual C++ 2015 Redistributable Update 3）

• 下一步安装CUDA和cuDNN：

可以看一下TensorFlow1.5版本build_info.py中的内容：

cudart_dll_name = 'cudart64_90.dll'
cuda_version_number = '9.0'
nvcuda_dll_name = 'nvcuda.dll'
cudnn_dll_name = 'cudnn64_7.dll'
cudnn_version_number = '7'

可见1.5版本是支持CUDA 9.0与cuDNN 7的，网上有人说1.5只支持CUDA8，怕是用了旧版本= =
下载安装包：
CUDA Toolkit 9.0 (Sept 2017)
cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0
注意版本千万要正确，之前我安装了CUDA9.1后会有报错信息：
ModuleNotFoundError: No module named '_pywrap_tensorflow_internal'
折腾了半天才发现原因出在哪里……

CUDA 9的安装包直接点击安装即可，至于cuDNN 7的需要解压后与CUDA 9的安装目录（通常是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0）合并，之后要保证以下四个目录添加到了%PATH%变量中，之后重启系统：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include

当然也可以使用Anaconda直接安装TensorFlow，然而Anaconda之前被我玩坏了，所以这里暂时不写这个的安装方式，具体可以去TensorFlow的官网教程看：https://www.tensorflow.org/install/install_windows，CPU版本同样不再阐述

• 最后测试是否安装成功：

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

• 测试信息：

C:\Users\Lenovo>python
Python 3.6.3 (v3.6.3:2c5fed8, Oct 3 2017, 18:11:49) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
b'Hello, TensorFlow!'

[hischen]

很有用的配置过程