fix horovod

tensorflow · Mar 15, 2024 · ec11220 · ec11220
1 parent 8c1081b
commit ec11220
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tools/docker/install/install_horovod.sh b/tools/docker/install/install_horovod.sh
@@ -17,7 +17,6 @@
 ## Select version.
 HOROVOD_VERSION=$1
 
-
 # Install horovod
 if [[ "$2" != "--only-cpu" ]]; then
   ln -sf /usr/include/nccl.h /usr/local/include/nccl.h
@@ -29,12 +28,12 @@ if [[ "$2" != "--only-cpu" ]]; then
   HOROVOD_WITHOUT_MXNET=1 \
   HOROVOD_WITH_MPI=1 \
   HOROVOD_WITHOUT_GLOO=1 \
-  python -m pip install --no-cache-dir --use-pep517 horovod==$HOROVOD_VERSION
+  python -m pip install --no-cache-dir horovod==$HOROVOD_VERSION
 else
   HOROVOD_WITH_TENSORFLOW=1 \
   HOROVOD_WITHOUT_PYTORCH=1 \
   HOROVOD_WITHOUT_MXNET=1 \
   HOROVOD_WITH_MPI=1 \
   HOROVOD_WITHOUT_GLOO=1 \
-  python -m pip install --no-cache-dir --use-pep517 horovod==$HOROVOD_VERSION
+  python -m pip install --no-cache-dir horovod==$HOROVOD_VERSION
 fi