笔记本是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等之前，否则不起作用。
转载于:https://www.cnblogs.com/excellent-ship/p/8604941.html
相关资源：数据结构—成绩单生成器
