在tensorflow中设置为cuda在python中使用带有cpu的tensorflow中的keras

时间:2018-05-27 13:46:57

标签: python tensorflow cuda keras cpu

我在cuda环境中运行ML所有gpu。

对于其他任务,我需要使用cpu来操作tensorflow。

当我在没有任何设置的情况下运行它时,它默认运行为GPU并导致以下错误。

  

tensorflow.python.framework.errors_impl.InternalError:无法创建会话。

使用所有GPU可能会发生此错误。

是否有可以在CPU的tensorflow中运行keras的命令?

0 个答案:

没有答案