查看keras或者tensorflow正在使用的GPU

本文主要是介绍查看keras或者tensorflow正在使用的GPU，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

查看keras或者tensorflow正在使用的GPU

1.查看当前可以使用的GPU

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

运行结果如下

['/job:localhost/replica:0/task:0/device:GPU:0','/job:localhost/replica:0/task:0/device:GPU:1']

2.查看当前设备的详细信息

from tensorflow.python.client import device_lib
import tensorflow as tf

这篇关于查看keras或者tensorflow正在使用的GPU的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---