限制TensorFlow只在CPU上運行的方法

　　筆記本是NVIDIA GeForce 940M的顯卡，只有2G的顯存，運行TensorFlow代碼時候常出現OOM(Out of Memory)的錯誤，原因是batch_size設置得太大導致顯存不足。如果想讓代碼僅僅運行在CPU下，可在原代碼中加入如下代碼：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

 　　注：上述代碼一定要放在import tensorflow或keras等之前，否則不起作用。