Add cuda*

ucsd-ets · Jan 9, 2025 · 43ad5b0 · 43ad5b0
1 parent b406971
commit 43ad5b0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -43,7 +43,10 @@ RUN mamba install -c rapidsai-nightly -c conda-forge jupyterlab-nvdashboard
 RUN mamba install -c "nvidia/label/cuda-12.1.1" \
     cuda-nvcc \
     cuda-version=$CUDA_VERSION \
+    cuda-toolkit=$CUDA_VERSION \
     nccl \
+    cudnn \
+    libcublas \
     -y && \
     fix-permissions $CONDA_DIR && \
     fix-permissions /home/$NB_USER && \
@@ -96,4 +99,4 @@ RUN ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer_plugi
     ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.8 /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.$TENSORRT_VERSION
 
 # Run datahub scripts
-RUN . /tmp/activate.sh
+RUN . /tmp/activate.sh