Fix Nvidia Image build

aws · Oct 8, 2024 · 7d49734 · 7d49734
1 parent df5df8e
commit 7d49734
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/e2e2/test/images/nvidia/Dockerfile b/e2e2/test/images/nvidia/Dockerfile
@@ -62,7 +62,8 @@ RUN cd /tmp \
     /var/lib/apt/lists/*
 
 # Install NCCL
-ARG NCCL_VERSION=2.22.3-1+cuda${CUDA_MAJOR_VERSION}.${CUDA_MINOR_VERSION}
+ARG NCCL_VERSION="2.22.3-1+cuda${CUDA_MAJOR_VERSION}.${CUDA_MINOR_VERSION}"
+
 RUN apt update \
   && apt install -y \
     libnccl2=${NCCL_VERSION}  \