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

Question

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

Answer 1

你 select cpu 通过像这样显式指定设备：

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