Tensorflow
DarrenXf
这个作者很懒,什么都没留下…
展开
-
Ubuntu 16.04 安装 CUDA 9.0 cuDNN 7.0 tensorflow-gpu 1.8.0
默认系统装的是Ubuntu 16.04,并且安装了显卡驱动。 走过很多坑,这些软件之间是互相依赖的,所以我们只安装这些固定的版本。 CUDA 9.0 cuDNN 7.0 tensorflow-gpu 1.8.0 希望以后会更容易安装些。 如果不是这篇文章写了很久,就照着这篇文章做,不要试图去安装这里涉及软件的最新版本。 第一步先安装 CUDA 9.0。这个并不是官方默认最新的。 下载地址...原创 2018-08-29 18:27:34 · 1893 阅读 · 0 评论 -
tensorflow 1.8 测试gpu代码
查看日志信息若包含gpu信息,就是使用了gpu。 下面的代码查看日志信息,如果包含gpu信息就是使用了gpu import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 输出 2018-08-30 13:06:50.335491: I tensorflow/c...原创 2018-08-30 13:11:15 · 3097 阅读 · 0 评论