判断gpu是否可用 作者: Dean0731 时间: 2023-03-25 分类: 默认分类 # 查看Keras 是否可使用gpufrom 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()) 标签: none
评论已关闭