1. # 查看主机的GPU
    2. gpus = tf.config.experimental.list_logical_devices(device_type="GPU")
    4. # 查看主机的CPU
    5. cpus = tf.config.experimental.list_logical_devices(device_type="CPU")
    7. # 设置当前程序可见的GPU范围
    8. gpus = tf.config.experimental.list_logical_devices(device_type="GPU")
    9. tf.config.set_visible_devices(devices=gpus[0:2], device_type="GPU")
    11. # 程序动态的获取显存
    12. gpus = tf.config.experimental.list_logical_devices(device_type="GPU")
    13. for gpu in gpus:
    14.     tf.config.experimental.set_memory_growth(device=gpu, true)
    16. # 程序获取固定显存大小 1GB
    17. gpus = tf.config.experimental.list_logical_devices(device_type="GPU")
    18. tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])