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

时间:2016-11-18 09:52:13

标签: python tensorflow driver gpu

我不确定这个问题是属于这里还是属于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语句中的字符串时没有错误。我有一台运行在Linux Mint 17.3上的AMD Radeon HD 6650M GPU,安装了fglrx-updates驱动程序(开源的没有工作,当xserver-xorg-ati是xserver-xorg-ati时,笔记本电脑会继续以软件渲染模式运行选择)。错误是否会发生,因为tensorflow无法使用fglrx驱动程序?这可以修复吗?有人在那里,在tensorflow上使用fglrx驱动程序没有这个问题吗?任何信息都有帮助。

1 个答案:

答案 0 :(得分:0)

不幸的是,目前tensorflow仅支持支持CUDA的NVIDIA GPU。

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