使用 fglrx 驱动程序在 GPU 上运行 tensorflow

Question

我不确定这个问题是属于这里还是属于 tensorflow 故障排除或 askubuntu，但我会从这里开始。我在 GPU 上运行 tensorflow 时遇到问题，例如以下代码：

import tensorflow as tf;

with tf.device('/gpu:0'):
    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)

给出输出： 设备映射：没有已知设备。

当我用“cpu:0”更改 with 语句中的字符串时没有错误。 I have a AMD Radeon HD 6650M GPU, running on Linux Mint 17.3, with fglrx-updates driver installed (the open-source one didn't work, the laptop would keep starting in software rendering mode when xserver-xorg-ati was selected) .是否会因为 tensorflow 无法使用 fglrx 驱动程序而发生错误？这可以解决吗？有没有人在没有这个问题的情况下在 tensorflow 上使用 fglrx 驱动程序？任何信息都有帮助。

Answer 1

很遗憾，目前 tensorflow 仅支持支持 CUDA 的 NVIDIA GPU。

您可以关注issue，跟踪支持开源 OpenCL 的工作。