From 7d49734cb14434f90c9838ec54b9057b18fa5eeb Mon Sep 17 00:00:00 2001
From: Sichao Wang <sichaow@amazon.com>
Date: Tue, 8 Oct 2024 18:51:26 +0000
Subject: [PATCH] Fix Nvidia Image build

---
 e2e2/test/images/nvidia/Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/e2e2/test/images/nvidia/Dockerfile b/e2e2/test/images/nvidia/Dockerfile
index db7ec5127..6c1e8b666 100644
--- 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}  \