Skip to content

Commit

Permalink
Fixed unit tests (flyteorg#789)
Browse files Browse the repository at this point in the history
* Fixed tests

Signed-off-by: Kevin Su <pingsutw@apache.org>

* Fixed tests

Signed-off-by: Kevin Su <pingsutw@apache.org>

* bump horovod and tensorflow versions

Signed-off-by: Samhita Alla <aallasamhita@gmail.com>

Co-authored-by: Samhita Alla <aallasamhita@gmail.com>
  • Loading branch information
pingsutw and samhita-alla authored Jun 7, 2022
1 parent 8759e27 commit 8941267
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cookbook/case_studies/ml_training/spark_horovod/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +73,10 @@ RUN sed -i 's/[ #]\(.*StrictHostKeyChecking \).*/ \1no/g' /etc/ssh/ssh_config &&
sed -i 's/#\(StrictModes \).*/\1no/g' /etc/ssh/sshd_config

# Horovod-related installations
RUN pip install tensorflow==2.7.0
RUN pip install tensorflow==2.9.1
# Enable GPU
# ENV HOROVOD_GPU_OPERATIONS NCCL
RUN HOROVOD_WITH_MPI=1 HOROVOD_WITH_TENSORFLOW=1 pip install --no-cache-dir horovod[spark,tensorflow]==0.23.0
RUN HOROVOD_WITH_MPI=1 HOROVOD_WITH_TENSORFLOW=1 pip install --no-cache-dir horovod[spark,tensorflow]==0.24.3

# Setup HOROVOD entrypoint
# TODO: Set the entrypoint
Expand Down

0 comments on commit 8941267

Please sign in to comment.