tf.config整体介绍

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