运行 带有 fglrx 驱动程序的 GPU 上的 tensorflow
Running tensorflow on GPU with fglrx driver
我不确定这个问题是属于这里还是属于 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 语句中的字符串时没有错误。我有一个 AMD Radeon HD 6650M GPU，运行 Linux Mint 17.3，安装了 fglrx-updates 驱动程序（开源驱动程序不起作用，笔记本电脑会在软件渲染模式下继续启动时选择了 xserver-xorg-ati）。是否会因为 tensorflow 无法使用 fglrx 驱动程序而发生错误？这可以修复吗？有没有人在没有这个问题的情况下在 tensorflow 上使用 fglrx 驱动程序？任何信息都有帮助。
遗憾的是，目前tensorflow只支持支持CUDA的NVIDIA GPU。
您可以关注 issue，跟踪支持开源 OpenCL 的努力。
我不确定这个问题是属于这里还是属于 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 语句中的字符串时没有错误。我有一个 AMD Radeon HD 6650M GPU，运行 Linux Mint 17.3，安装了 fglrx-updates 驱动程序（开源驱动程序不起作用，笔记本电脑会在软件渲染模式下继续启动时选择了 xserver-xorg-ati）。是否会因为 tensorflow 无法使用 fglrx 驱动程序而发生错误？这可以修复吗？有没有人在没有这个问题的情况下在 tensorflow 上使用 fglrx 驱动程序？任何信息都有帮助。
遗憾的是，目前tensorflow只支持支持CUDA的NVIDIA GPU。
您可以关注 issue，跟踪支持开源 OpenCL 的努力。