【python】Keras、Tensorflow 如何切换 GPU 和 CPU(强制使用CPU)

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

import os

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

你可能感兴趣的:(【python】Keras、Tensorflow 如何切换 GPU 和 CPU(强制使用CPU))