使用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等不同类型的设备。
