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