Windows10下安装TensorFlow 1.5-GPU版

搞了好几个小时终于配置好了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!'

hiyouga / hiyouga-blog-project

Windows10下安装TensorFlow 1.5-GPU版 #4