Use a GPU | TensorFlow Core
https://www.tensorflow.org/guide/gpu11.11.2021 · TensorFlow code, and tf.keras models will transparently run on a single GPU with no code changes required.. Note: Use tf.config.list_physical_devices('GPU') to confirm that TensorFlow is using the GPU. The simplest way to run on multiple GPUs, on one or many machines, is using Distribution Strategies.. This guide is for users who have tried these …