tensorflow,pytorch,paddlepaddle查看GPU是否可用

1、tensorflow

import tensorflow as tf
print(tf.test.is_gpu_available())

2、pytorch

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

3、paddlepaddle

import paddle
paddle.fluid.install_check.run_check()