原文链接:https://stackoverflow.com/questions/45662253/can-i-run-keras-model-on-gpu
- 查看是否有GPU可用
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
- 使用GPU
from keras import backend as K
K.tensorflow_backend._get_available_gpus()