TensorFlow2.0选择GPU或CPU训练

注意：以下所有程序都要在导入TensorFlow和keras包之前加入才可以

选择CPU：

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

选择GPU：

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

查看当前设备使用的是CPU还是GPU：

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())