Win10下tensorflow-gpu的配置

1、安装tensorflow-gpu

pip install tensorflow==1.14.0
pip install tensorflow-gpu==1.14.0

2、查看tensorflow-gpu对应版本的CUDA和cuDNN
查看对应版本的网址：https://tensorflow.google.cn/install/source_windows#gpu

3、安装CUDA
网址 https://developer.nvidia.com/cuda-toolkit-archive

4、下载cuDNN
官网 https://developer.nvidia.com/rdp/cudnn-archive
cuDNN解压后的文件

将这三个文件夹复制到下面这个路径

5、添加系统变量


6、检查自己的显卡驱动版本是否对应
官网链接 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

显卡驱动版本在GeForce Experience中查看

7、测试程序（在cmd中运行)

import tensorflow as tf
a = tf.random_normal((100,100))
b = tf.random_normal((100,500))
c = tf.matmul(a, b)#矩阵相乘
sess = tf.InteractiveSession()
sess.run(c)

如果出现以下显示说明配置成功：

注：参考了很多大佬的博客，就不例举了。