Dockerfile should use the correct nvidia channel to install libcu* (#423

)
NVIDIA · Dec 12, 2024 · f56e924 · f56e924
1 parent 1c0382e
commit f56e924
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -29,14 +29,16 @@ 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 \
+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 \
   libcublas \
   libcurand \
   libcusparse \
-  libcusolver && \
+  libcusolver \
+  cuda-nvvm && \
   source activate curator && \
   pip install --upgrade pytest pip