From ea4647f422baec8aa4d76c0dc6948fe03dfbb2dd Mon Sep 17 00:00:00 2001 From: Praateek Date: Wed, 11 Dec 2024 08:27:39 -0800 Subject: [PATCH 1/4] fc Signed-off-by: Praateek --- Dockerfile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 51fe7be4..e66936c0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,14 +29,15 @@ LABEL "nemo.library"=${IMAGE_LABEL} WORKDIR /opt # Install the minimal libcu* libraries needed by NeMo Curator -RUN conda create -y --name curator -c conda-forge -c nvidia \ +RUN conda create -y --name curator -c -c nvidia/label/cuda-${CUDA_VER} -c conda-forge python=3.10 \ cuda-cudart \ libcufft \ libcublas \ libcurand \ libcusparse \ - libcusolver && \ + libcusolver \ + cuda-nvvm && \ source activate curator && \ pip install --upgrade pytest pip From 3a3d547c4715195ef5006ae4db5e7656c48b4ed3 Mon Sep 17 00:00:00 2001 From: Praateek Date: Wed, 11 Dec 2024 08:31:50 -0800 Subject: [PATCH 2/4] remove extra -c Signed-off-by: Praateek --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index e66936c0..937ba999 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,7 +29,7 @@ LABEL "nemo.library"=${IMAGE_LABEL} WORKDIR /opt # Install the minimal libcu* libraries needed by NeMo Curator -RUN conda create -y --name curator -c -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 \ From 586701aca551393f4c9287c8620489cc7eca2b09 Mon Sep 17 00:00:00 2001 From: Praateek Date: Wed, 11 Dec 2024 10:32:03 -0800 Subject: [PATCH 3/4] missing / Signed-off-by: Praateek --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 937ba999..619c3d1f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,7 +29,7 @@ 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 +RUN conda create -y --name curator -c nvidia/label/cuda-${CUDA_VER} -c conda-forge \ python=3.10 \ cuda-cudart \ libcufft \ From a2ed49f530b03be83e598a5d6312d13db005415c Mon Sep 17 00:00:00 2001 From: Praateek Date: Wed, 11 Dec 2024 11:16:48 -0800 Subject: [PATCH 4/4] use env var? Signed-off-by: Praateek --- 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 \