Tensorflow/Keras 指定CPU运行

全局配置 运行TensorFlow代码时候常出现OOM(Out of Memory)的错误，缘由是batch_size设置得太大致使显存不足。若是想让代码仅仅运行在CPU下，可在原代码中加入以下代码：python import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]

>>阅读原文<<