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"]

>>阅读原文<<