在CPU和GPU的两个单独的Juypter笔记本中训练Keras模型

时间:2019-11-06 12:22:42

标签: python keras gpu cpu

我正在Jupyter Notebook上针对两种不同应用训练Keras CNN模型。考虑到我想利用PC的全部资源,我可以在一个笔记本电脑中使用Keras-GPU,而在另一个使用CPU的笔记本电脑中使用吗?

我了解到Keras默认情况下使用GPU(如果有),并且我可以强迫Keras将CPU用作  在Can Keras with Tensorflow backend be forced to use CPU or GPU at will?中。我的问题是,通过运行以下代码,



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

默认设置会在所有正在运行的笔记本电脑中还是仅在特定笔记本电脑中更改?

1 个答案:

答案 0 :(得分:1)

通过运行以下代码行,

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

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

您可以使用

os.environ['CUDA_VISIBLE_DEVICES'] = ''

在CPU上训练

相关问题