From a2ed49f530b03be83e598a5d6312d13db005415c Mon Sep 17 00:00:00 2001
From: Praateek <praateekm@gmail.com>
Date: Wed, 11 Dec 2024 11:16:48 -0800
Subject: [PATCH] use env var?

Signed-off-by: Praateek <praateekm@gmail.com>
---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 619c3d1f..37bcb35d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -29,7 +29,8 @@ LABEL "nemo.library"=${IMAGE_LABEL}
 WORKDIR /opt
 
 # Install the minimal libcu* libraries needed by NeMo Curator
-RUN conda create -y --name curator -c nvidia/label/cuda-${CUDA_VER} -c conda-forge \
+ENV _CUDA_VER=${CUDA_VER}
+RUN conda create -y --name curator -c nvidia/label/cuda-${_CUDA_VER} -c conda-forge \
   python=3.10 \
   cuda-cudart \
   libcufft \