tf.config整体介绍
- 物理设备对应
PhysicalDevice,逻辑设备对应LogicalDevice。
list_physical_devices 函数可以列出主机上所有的物理设备。list_logical_devices函数可以列出运行时所有的逻辑设备。
set_visible_devices函数指定哪些物理设备在运行时可见。
- 默认情况下,运行时初始化后,可见的
PhysicalDevice会有一个LogicalDevice与之关联。
set_logical_device_configuration函数可以在运行时初始化前在同一个物理设备上创建多个逻辑设备,其中逻辑设备的配置由LogicalDeviceConfiguration对象指定
LogicalDeviceConfiguration对象指定逻辑设备的配置,主要有三个参数:内存限制、优先级(用于任务调度)、序号(用于排序)。
- 一般在运行时,如果我们指定的设备不存在会报错,
set_soft_device_placement 函数可以开启软设备放置,在指定的设备不存在时自动切换到其他设备。
get_logical_device_configuration函数可以获取物理设备的虚拟配置。get_soft_device_placement函数可以查看当前软设备放置是否开启。