【python】Keras、Tensorflow 如何切换 GP

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

import os

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