时间戳

首页 > 解决方案 > 在 CPU 和 GPU 上的两个单独的 juypter 笔记本中训练 Keras 模型

python - 在 CPU 和 GPU 上的两个单独的 juypter 笔记本中训练 Keras 模型

问题描述

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

我了解到 Keras 默认使用 GPU（如果可用），我可以强制 Keras 使用 CPU，就像在Can Keras with Tensorflow backend 是否可以随意使用 CPU 或 GPU 中一样？. 我的问题是，通过运行这行代码，



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

是否会在所有正在运行的笔记本或仅在该特定笔记本中更改默认设置？

标签： pythonkerasgpucpu

解决方案

通过运行这行代码，

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

仅在该特定笔记本中更改默认设置

您可以使用

os.environ['CUDA_VISIBLE_DEVICES'] = ''

在 CPU 上训练

推荐阅读