mofed

mosaicml · May 20, 2024 · 9300331 · 9300331
1 parent 28bc538
commit 9300331
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -293,6 +293,8 @@ RUN if [ -n "$AWS_OFI_NCCL_VERSION" ] ; then \
 # Mellanox OFED driver installation
 ###################################
 
+ARG MOFED_VERSION
+
 RUN if [ -n "$MOFED_VERSION" ] ; then \
         if [ "$TARGETARCH" = "amd64" ]; then \
             MOFED_PACKAGE="x86_64"; \