怎么查看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]:

[‘/job:localhost/replica:0/task:0/device:GPU:0‘]

查看更详细device信息

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

print(device_lib.list_local_devices())
print(tf.test.is_built_with_cuda())

output:

 [name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality 

incarnation: 9443078288448539683
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality 

incarnation: 14589028880023685106
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality 

incarnation: 12944586764183584921
physical_device_desc: "device: XLA_GPU device"
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 8365150044
locality 
  bus_id: 1
  links 
  

incarnation: 8725535454902618392
physical_device_desc: "device: 0, name: Tesla P100-PCIE-16GB, pci bus id: 0000:00:04.0, compute capability: 6.0"
]
True
 
查看正在使用的GPU
import tensorflow as tf
print (tf.__version__)
if tf.test.gpu_device_name():
    print(Default GPU Device: .format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")
 
 

output: 

1.13.1
Default GPU Device: /device:GPU:0
 

 

以上是关于怎么查看keras 或者 tensorflow 正在使用的GPU的主要内容,如果未能解决你的问题,请参考以下文章

tensorflow报错 或者 keras报错 以及tf.keras报错: OOM 显存不足

标准化 Keras:TensorFlow 2.0 高级 API 指南

Tensorflow 与 Keras 中的 RNN,tf.nn.dynamic_rnn() 的贬值

Keras TensorFlow 2.0 精华资源

TensorFlow 和 Keras 中的符号张量是啥?

Tensorflow (.pb) 格式到 Keras (.h5)