From ff623f9dc3fcc32f26200a9984823a1f8c93ffa7 Mon Sep 17 00:00:00 2001 From: Veronika Scholz <veronika.scholz@tu-dresden.de> Date: Thu, 7 Oct 2021 12:40:36 +0200 Subject: [PATCH] Update distributed_training.md --- doc.zih.tu-dresden.de/docs/software/distributed_training.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 1ae949c83..c4e7666d0 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() -- GitLab