在 CPU 和 GPU 上的两个独立的 jupyter 笔记本中训练 Keras 模型

Question

我正在为 Jupyter Notebook 上的两个不同应用程序训练 Keras CNN 模型。鉴于我想利用我 PC 的全部资源，我可以在一个笔记本上使用 Keras-GPU，在另一个笔记本上使用 CPU。

我了解到 Keras 默认使用 GPU - 如果可用 - 我可以强制 Keras 使用 CPU 作为 在 。我的问题是，通过 运行 这行代码，

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

默认设置会在所有 运行 笔记本中更改还是仅在特定笔记本中更改？

Answer 1

通过运行这行代码，

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

仅特定笔记本中的默认设置发生变化

您可以使用

os.environ['CUDA_VISIBLE_DEVICES'] = ''

在 CPU

上训练