diff --git a/doc.zih.tu-dresden.de/docs/software/distributed_training.md b/doc.zih.tu-dresden.de/docs/software/distributed_training.md index 1ae949c838b5c358c567c2f57805f0fd5f3abbfc..c4e7666d0b226a7d6691b76c3221265a85982a82 100644 --- a/doc.zih.tu-dresden.de/docs/software/distributed_training.md +++ b/doc.zih.tu-dresden.de/docs/software/distributed_training.md @@ -269,8 +269,8 @@ To have better performance it is recommended to use NCCL instead of OpenMPI. ##### Verify that Horovod works ```python -import torch #import pytorch -import horovod.torch as hvd #import horovod +import tensorflow #import pytorch +import horovod.tensorflow as hvd #import horovod hvd.init() #initialize horovod hvd.size() hvd.rank()