use env var?

Signed-off-by: Praateek <[email protected]>
NVIDIA · Dec 11, 2024 · a2ed49f · a2ed49f
1 parent 586701a
commit a2ed49f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 \