判断gpu是否可用
Posted dean0731
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了判断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())
以上是关于判断gpu是否可用的主要内容，如果未能解决你的问题，请参考以下文章