判断gpu是否可用
# 查看Keras 是否可使用gpu
from keras import backend as K gpu =K.tensorflow_backend._get_available_gpus() print(gpu)
# 查看tensorflow是否可使用gpu import tensorflow as tf print(tf.test.is_gpu_available())
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接,如有问题, 可评论咨询.