Poli
Poli

Reputation: 1

Will TensorFlow use multiple GPU's by default?

When I do not allocate the device using "with tf.device(d):", will TensorFlow use only one GPU or will it use all the available GPUs when one GPU is not enough?

Upvotes: 0

Views: 211

Answers (1)

Ortomala Lokni
Ortomala Lokni

Reputation: 62733

The TensorFlow documentation says:

If you have more than one GPU in your system, the GPU with the lowest ID will be selected by default. If you would like to run on a different GPU, you will need to specify the preference explicitly.

Upvotes: 2

Related Questions