From 35b5993bd904bfd13da59fcb6a124a0a4af6a0e2 Mon Sep 17 00:00:00 2001
From: Praateek Mahajan <praateekmahajan@users.noreply.github.com>
Date: Mon, 23 Dec 2024 02:16:50 -0800
Subject: [PATCH] Bug fix in dockerfile ARG vs ENV var (#446)

---
 Dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 37bcb35d..b0d1bedc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,9 +28,11 @@ FROM rapidsai/ci-conda:cuda${CUDA_VER}-${LINUX_VER}-py${PYTHON_VER}
 LABEL "nemo.library"=${IMAGE_LABEL}
 WORKDIR /opt
 
+# Re-declare ARGs after new FROM to make them available in this stage
+ARG CUDA_VER
+
 # Install the minimal libcu* libraries needed by NeMo Curator
-ENV _CUDA_VER=${CUDA_VER}
-RUN conda create -y --name curator -c nvidia/label/cuda-${_CUDA_VER} -c conda-forge \
+RUN conda create -y --name curator -c nvidia/label/cuda-${CUDA_VER} -c conda-forge \
   python=3.10 \
   cuda-cudart \
   libcufft \