TensorFlow 强制使用CPU 因为显存只有8G,训练时各种报错。有时训练几次就崩了。所以直接使用cpu,有条件的还是使用gpu的快。 import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 强制使用cpu