Skip to content

Commit da90f41

Browse files
authored
Update Dockerfile with required packages for perf testing (#806)
Install hf-hub
1 parent 8da2012 commit da90f41

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

3.test_cases/megatron/nemo/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,9 @@ RUN rm -rf /var/lib/apt/lists/*
106106
RUN echo "hwloc_base_binding_policy = none" >> /opt/amazon/openmpi/etc/openmpi-mca-params.conf \
107107
&& echo "rmaps_base_mapping_policy = slot" >> /opt/amazon/openmpi/etc/openmpi-mca-params.conf
108108

109+
# Install additional dependencies required for perf testing
110+
RUN pip3 install --upgrade huggingface_hub>=0.34.0
111+
109112
RUN pip3 install awscli pynvml wandb
110113

111114
RUN mv $OPEN_MPI_PATH/bin/mpirun $OPEN_MPI_PATH/bin/mpirun.real \

0 commit comments

Comments
 (0)