win10 安装 GPU版TensorFlow

WIN10安装CUDA9+CUDNN+TENSORFLOW1.9.0

安装环境：笔记本电脑win10专业版操作系统 + 1050显卡

1、安装CUDA9

地址：https://developer.nvidia.com/cuda-90-download-archive

安装时会先选择解压位置作为临时存放，可以自己定义。

安装过程中可能会弹出以下警告：

这个是由于安装的显卡驱动版本高于CUDA9所自带的驱动版本号，点击继续即可。安装的时候建议选择自定义,不选VS的组件：

自己可以选择路径安装，结束。

2、下载安装CUNNN

下载cudnn,下载地址：https://developer.nvidia.com/cudnn，进去之后,找到对应的版本，点击Download，他会要你注册登录，然后填个问卷，才可以下载，这一步自行按照网站提示做就可以了。提交完问卷后，就可以点击下载了。

选择以下的版本均可。

下载完成后解压，把解压后的3个文件夹放入CUDA9安装对应的文件夹中。

这里我的CUDA9路径是自己定义的。

3、安装tensorflow-gpu V1.9.0

tensorflow-gpu 版本过高运行程序时，会报错。实测V1.11.0也是可以正常使用的。

pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==1.9.0

4.检测是否成功

打开任意一个python编辑环境，输入以下代码

import tensorflow as tf

hello = tf.constant(‘hello,TensorFlow!’)

sess = tf.Session()

print(sess.run(hello))

最后一行显示 b’hello，TensorFlow!'说明Tensorflow配置成功！！

注意：这里安装的是TensorFlow1.x版本，若要安装2.x版本，方法相同，只是换为CUDA10+CUDNN+TensorFlow2.0