tensorflow - 如何配置 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，并且如果可能的话，也使这个配置永久用于我以前/未来的所有程序？

标签： tensorflowgpuconfig

Answer 1

尝试使用以下功能：set_visible_devices

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

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