测试GPU是否可用

测试安装的GPU驱动是否成功，或者深度学习框架是否可用。

tensorflow/keras

>>> import tensorflow as tf
# 对于tf1.x版本，如果输出True，则表示GPU可以使用，False表示驱动安装不正确或者其他原因
>>> print(tf.test.is_gpu_available())
True

# 对于tf2.5版本，可以用下面语句输出可用的哪几块GPU，其他的2.1~2.4没测试
>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU'),  
 PhysicalDevice(name='/physical_device:GPU:1', device_type='GPU'),  
 PhysicalDevice(name='/physical_device:GPU:2', device_type='GPU'),  
 PhysicalDevice(name='/physical_device:GPU:3', device_type='GPU')]

pytorch

>>> import torch
# 如果输出True，则表示GPU可以使用，False表示驱动安装不正确或者其他原因
>>> print(torch.cuda.is_available())

paddlepaddle

>>> import paddle
# 如果输出PaddlePaddle is installed successfully!，则表示GPU可以使用
>>> paddle.utils.run_check()