测试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()

如果输出以下结果说明调用成功：

Done!!!