怎么查看keras 或者 tensorflow 正在使用的GPU
Posted mashuai-191
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了怎么查看keras 或者 tensorflow 正在使用的GPU相关的知识,希望对你有一定的参考价值。
查看keras认得到的GPU
from keras import backend as K K.tensorflow_backend._get_available_gpus()
Out[28]:
以上是关于怎么查看keras 或者 tensorflow 正在使用的GPU的主要内容,如果未能解决你的问题,请参考以下文章
tensorflow报错 或者 keras报错 以及tf.keras报错: OOM 显存不足
标准化 Keras:TensorFlow 2.0 高级 API 指南