TensorFlow

API

 tf.config / experimental_list_devices


Get if soft device placement is enabled.

If enabled, an op will be placed on CPU if any of the following are true

  1. there's no GPU implementation for the OP
  2. no GPU devices are known or registered
  3. need to co-locate with reftype input(s) which are from CPU

If soft placement is enabled.

此页内容是否对您有帮助