查看keras或者tensorflow正在使用的GPU
1.查看当前可以使用的GPU
from keras import backend as K
K.tensorflow_backend._get_available_gpus()
运行结果如下
['/job:localhost/replica:0/task:0/device:GPU:0',
'/job:localhost/replica:0/task:0/device:GPU:1']
2.查看当前设备的详细信息
from tensorflow.python.client import device_lib
import tensorflow as tf