检查tensorflow gpu 或keras gpu 或pytorch gpu是否安装成功

tensorflow gpu：

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

keras gpu：

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

pytorch gpu

import torch
print(torch.cuda.is_available())