Win10 tensorflow-gpu安装

Win10 tensorflow-gpu安装

1. 搭建win10+python3.6+vs2017+cuda10.0+pycuda开发环境

参见上一篇博客.

上一篇cuda选择的是10.1版本，但是，目前没有找到和10.1匹配的cudnn和tensorflow-gpu版本，所以笔者又重装了cude，改用10.0版本，这里注意环境变量也要修改。

2. 安装cudnn

cudnn下载地址.
我选择的版本是7.4.2

下载完成，解压后将bin,include,lib文件夹里的内容分别拷到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0目录下的bin,include,lib文件夹里面，注意不是拷贝整个文件夹而是里面的文件。
拷贝完成之后，将bin目录和lib/x64目录添加到环境变量的path中，不然会报错找不到dll文件。

3.安装tenserflow-gpu

pip3 install tenserflow-gpu，当前版本是1.14.0

4.测试

运行如下代码：

import tensorflow as tf

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
def f(x) :
    return c
#tensorflow2.x 取消了session
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

print(sess.run(c))

输出以下结果说明安装成功

可能会出现的问题

1）

2)

出现以上问题首先确定第二步骤中的环境变量是否添加，如果添加了就是因为版本不匹配，注意tensorflow-gpu，cudnn和cuda的版本对应关系。

具体对应关系可参考下面的链接：
https://blog.csdn.net/qq_27825451/article/details/89082978