一、安装环境

1.1 操作系统：Windows 10 64位

1.2 显卡：NVIDIA 965M（Compute Capability 5.2）CUDA查询链接

1.3 NVIDIA Driver 416.34

1.4 CUDA Toolkit 9.0 Base Installer & Patch1-4

1.5 cuDNN v7.3.1

1.6 TensorFlow-GPU 1.11

二、安装过程

2.1 安装CUDA

2.1.1 安装CUDA Toolkit 9.0 Base Installer

2.1.2 安装CUDA Patch1-4

其中，安装Patch1时程序闪退，解决方案：删除C:\Program Files\NVIDIA Corporation\Installer2文件夹后重新安装。

2.1.3 设置环境变量

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0 
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 
CUDA_BIN_PATH = %CUDA_PATH%\bin 
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64 

2.1.4 测试环境变量

set cuda

2.2 安装cuDNN v7.3.1 过程参照官方文档

安装完后测试CUDA。
进入 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite 目录下执行 deviceQuery.exe

执行 bandwidthTest.exe

返回值为 PASS 说明安装成功

2.3 安装TensorFlow

2.3.1 创建虚拟环境

conda create -n TensorFlow-GPU python=3.6

2.3.2 进入虚拟环境

activate TensorFlow-GPU

2.3.3 安装GPU版TensorFlow

pip install --ignore-installed --upgrade tensorflow-gpu

2.3.4 测试TensorFlow

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

参考链接：

https://blog.csdn.net/dou3516/article/details/77836459
https://devtalk.nvidia.com/default/topic/1039261/cuda-9-0-patch-1-installation-problem/
https://blog.csdn.net/u013165921/article/details/77891913