如何使Spyder代码在GPU上而不是Ubuntu上的cpu上运行？

Question

我正在Spyder上创建具有GPU的机器上的深度学习模型，我发现该机器在CPU上工作并且我的代码运行了很长时间。首先我下载了​​tensorflow-GPU，但我不知道如何开始在GPU上工作。如何使我的Spyder代码在GPU上而不是Ubuntu上的cpu上运行？

Answer 1

您可以通过显式指定设备来选择cpu：

with tf.device("cpu"):
    with tf.Session() as sess:
         your code