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

Question

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

[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"
]

我想要做的是将我的程序配置为使用GeForce GTX 960M并在可能的情况下使此配置永久用于我以前/未来的所有程序？

Answer 1

尝试使用函数： set_visible_devices

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

在哪里可以指定要使用的 GPU