Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
zhenghh04 authored Jan 23, 2023
1 parent 50312a7 commit ca901ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion 06_distributedTraining/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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')
```
Expand Down

0 comments on commit ca901ae

Please sign in to comment.