欢迎访问宙启技术站
智能推送

使用list_local_devices()方法获取本地设备列表的方法

发布时间:2023-12-18 02:24:48

要使用list_local_devices()方法获取本地设备列表,首先需要导入tensorflow库,并调用tf.config.list_local_devices()方法。该方法将返回一个列表,其中包含本地计算设备的相关信息。

以下是一个使用list_local_devices()方法的例子:

import tensorflow as tf

# 调用list_local_devices()方法获取本地设备列表
devices = tf.config.list_local_devices()

# 打印设备列表信息
for device in devices:
    print(f"设备名称: {device.name}")
    print(f"设备类型: {device.device_type}
")

这段代码首先导入tensorflow库,然后调用list_local_devices()方法获取本地设备列表,并将返回的结果存储在devices变量中。

接下来,使用一个循环遍历devices列表,并打印每个设备的名称和设备类型。

运行以上代码,你将会得到一个输出类似下面的设备列表信息:

设备名称: /physical_device:CPU:0
设备类型: CPU

设备名称: /physical_device:GPU:0
设备类型: GPU

上面输出的设备名称中的CPU:0表示 个CPU设备,GPU:0表示 个GPU设备。

注意:list_local_devices()方法返回的设备列表中可能包含CPU、GPU等不同类型的设备。