判断gpu是否可用

代码语言：javascript复制

# 查看Keras 是否可使用gpu
from keras import backend as K
gpu =K.tensorflow_backend._get_available_gpus()
print(gpu)

代码语言：javascript复制

# 查看tensorflow是否可使用gpu
import tensorflow as tf
print(tf.test.is_gpu_available())

gpu

0 人点赞