Keras tensorflow后端不检测GPU

时间:2017-07-18 15:54:09

标签: tensorflow keras tensorflow-gpu

我在linux上使用tensorflow后端运行keras。 首先,我自己安装了tensorflow GPU版本,并运行以下代码进行检查,发现它在GPU上运行并显示正在运行的GPU,设备映射等。我使用的张量流来自UPDATE APUI SET APUI.udTempReviewer = APUL.ReviewerGroup FROM APUI INNER JOIN APUL ON APUI.UISeq = APUL.UISeq AND APUI.APCo = APUL.APCo < / p>

https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-
0.11.0-cp27-none-linux_x86_64.whl

然后,我使用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) # Creates a session with log_device_placement set to True. sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) # Runs the op. print(sess.run(c)) 安装了keras。我检查了conda install keras,现在我有两个版本的tensorflow(1.1.0和0.11.0)。我尝试了conda list,结果是:

import tensorflow as tf

我已经设置了2017-07-18 16:35:59.569535: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations. 2017-07-18 16:35:59.569629: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations. 2017-07-18 16:35:59.569690: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations. 2017-07-18 16:35:59.569707: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX2 instructions, but these are available on your machine and could speed up CPU computations. 2017-07-18 16:35:59.569731: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations. Device mapping: no known devices. 2017-07-18 16:35:59.579959: I tensorflow/core/common_runtime/direct_session.cc:257] Device mapping: MatMul: (MatMul): /job:localhost/replica:0/task:0/cpu:0 2017-07-18 16:36:14.369948: I tensorflow/core/common_runtime/simple_placer.cc:841] MatMul: (MatMul)/job:localhost/replica:0/task:0/cpu:0 b: (Const): /job:localhost/replica:0/task:0/cpu:0 2017-07-18 16:36:14.370051: I tensorflow/core/common_runtime/simple_placer.cc:841] b: (Const)/job:localhost/replica:0/task:0/cpu:0 a: (Const): /job:localhost/replica:0/task:0/cpu:0 2017-07-18 16:36:14.370109: I tensorflow/core/common_runtime/simple_placer.cc:841] a: (Const)/job:localhost/replica:0/task:0/cpu:0 ,它在安装keras之前有效。 这是因为tensorflow版本?安装keras时,我可以选择安装0.11.0而不是1.1.0吗? 如果问题是由于张量流没有检测到GPU,我该如何解决这个问题?我在这篇link中读到了它,并说它会在GPU上自动运行,因为它会检测到一个。

1 个答案:

答案 0 :(得分:3)

有可能Keras(取决于更新版本的TensorFlow)导致安装了仅限CPU的TensorFlow软件包(tensorflow),该软件包隐藏了较旧的支持GPU的版本(tensorflow-gpu )。

我会首先升级支持GPU的版本。通常您只需执行pip install --upgrade tensorflow-gpu,但TensorFlow installation page中有特定于Anaconda的说明。然后,您可以使用pip uninstall tensorflow卸载仅限CPU的TensorFlow软件包。现在import tensorflow as tf应该实际导入支持GPU的软件包,正如您所建议的那样,它应该自动检测您的GPU。