PyTorch、Keras、TensorFlow、MXNet等深度学习框架测试能否使用GPU

1、检查电脑可使用设备

先检查自己有没有gpu

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

以我自己电脑为例，输出如下结果。

2、测试PyTorch GPU

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

3、测试Keras GPU

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

4、测试TensorFlow GPU

import tensorflow as tf

print(tf.test.is_built_with_cuda())
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

5、测试 MXNet GPU

import mxnet as mx
print(mx.gpu())