Windows10下安装Tensorflow-GPU你可以避免的坑

Windows10下安装Tensorflow-GPU你可以避免的坑

工欲善其事必先利其器。有人说,tensorflow-gpu安装好烦躁哦,各种兼容性问题奥,怎么搞?

问题 1

如何查看自己的CUDA驱动适配版本?

回答

桌面右键打开英伟达控制面板，点击帮助->系统信息->组件。

问题2

我该下载哪个版本的CUDA和cudnn呢?

回答

问题3

我就头铁,用python3.6安装tensorflow-gpu 1.14.0版本,这样会报错吗?

回答

会。会斩钉截铁的告诉你缺少XX模块。

问题 4

我该如何测试我的CUDA系列环境安装成功了?

回答

问题5

我的电脑有点牛鼻,可以降级CUDA版本吗?

回答

可以。讲道理,是可以的。然后在tf-gpu版本还没跟上cuda时候,还是得降级,有时候还是有必要的。

问题6:

电脑正常安装tensorflow-gpu,但是我应该怎么去测试安装成功呢?

回答

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)


with tf.Session() as sess:
    print(sess.run([a,b]))

print(a+b)

print('GPU:', tf.test.is_gpu_available())

然后出现如下结果,说明安装时正常的。

问题7:

我想临时快速的下载一个包,怎么搞?

回答:

pip install keras -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

镜像源和package可以按照需要改动。

查找灵感

参考1
参考2
参考3
参考4
参考5
参考6