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

【问题标题】：How to make my Spyder code run on GPU instead of cpu on Ubuntu?如何让我的 Spyder 代码在 GPU 而不是 Ubuntu 上的 cpu 上运行？
【发布时间】：2020-04-13 16:54:58
【问题描述】：

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

【问题讨论】：

标签： python tensorflow ubuntu deep-learning

【解决方案1】：

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

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

【讨论】：

@Achai Gupta 感谢您的解决方案。您的意思是 tf.device("gpu"): with tf.Session() as sess: 我想在 GPU 上工作而不是 cpu
@Anchai Gupta 它不起作用。你能帮我如何在 GPU 上工作吗？？
你能上传输出看看错误在哪里吗？ 从 tensorflow.python.client 导入 device_lib device_lib.list_local_devices()
@Anchai Gupta 语法无效
从 tensorflow.python.client 导入 device_lib