diff --git a/06_distributedTraining/README.md b/06_distributedTraining/README.md index 705142eca..e75544427 100644 --- a/06_distributedTraining/README.md +++ b/06_distributedTraining/README.md @@ -99,7 +99,7 @@ Reference: https://horovod.readthedocs.io/en/stable/ gpus = tf.config.experimental.list_physical_devices('GPU') # Ping GPU to the rank for gpu in gpus: - tf.config.experimental.set_memory_growth(gpu, True) + tf.config.experimental.set_memory_growth(gpu, True) if gpus: tf.config.experimental.set_visible_devices(gpus[hvd.local_rank()], 'GPU') ```