在tensorflow中使用device_count {＆＃39; GPU＆＃39;：1,2,3}时语法无效

Question

所有

当我尝试让tensorflow只使用特定的GPU时遇到一个SyntaxError，这里是代码和错误信息，有谁知道如何解决这个问题？非常感谢！

here is the code

err info

Answer 1

您提供的代码表明您有语法错误。

要在GPU上手动运行代码，您可以执行以下操作：

with tf.device('/device:GPU:1'):
  ##your code

如果您的目标是在多个GPU上运行代码，请尝试以下方法：

for d in ['/device:GPU:2', '/device:GPU:3']:
  with tf.device(d):
    ##your code

您可以在此处找到更多信息：https://www.tensorflow.org/programmers_guide/using_gpu