我不确定这个问题是属于这里还是属于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驱动程序没有这个问题吗?任何信息都有帮助。