TensorFlow进行GPU和CPU切换(强制使用CPU)

在代码的开头，所有 import 之前，加入下面两行代码：

import os
["CUDA_VISIBLE_DEVICES"] = "-1"  # 这一行注释掉就是使用gpu，不注释就是使用cpu