We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8da2012 commit da90f41Copy full SHA for da90f41
3.test_cases/megatron/nemo/Dockerfile
@@ -106,6 +106,9 @@ RUN rm -rf /var/lib/apt/lists/*
106
RUN echo "hwloc_base_binding_policy = none" >> /opt/amazon/openmpi/etc/openmpi-mca-params.conf \
107
&& echo "rmaps_base_mapping_policy = slot" >> /opt/amazon/openmpi/etc/openmpi-mca-params.conf
108
109
+# Install additional dependencies required for perf testing
110
+RUN pip3 install --upgrade huggingface_hub>=0.34.0
111
+
112
RUN pip3 install awscli pynvml wandb
113
114
RUN mv $OPEN_MPI_PATH/bin/mpirun $OPEN_MPI_PATH/bin/mpirun.real \
0 commit comments