如何配置Tensorflow以使用特定的GPU？

Question

这些是我拥有的已激活设备：

  render() {
    return (
      <div>
        <Grid item xs={7} style={{ marginTop: 32 }}>
          Condition 1
        </Grid>
        {this.createRadioGroups()}
      </div>
    );
  }

我想做的是将程序配置为使用[name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 5415837867258701517 , name: "/device:GPU:0" device_type: "GPU" memory_limit: 3198956339 locality { bus_id: 1 links { } } incarnation: 12462133041849407996 physical_device_desc: "device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0, compute capability: 5.0" ] ，并在可能的情况下使此配置永久存在于我以前的所有/将来的程序中？

Answer 1

尝试使用以下功能：set_visible_devices

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.set_visible_devices(physical_devices[1:],'GPU')

您可以在此处指定要使用的GPU