From 43ad5b08520dd5ee020a83f776d756be5518579a Mon Sep 17 00:00:00 2001
From: Dominic Feliton <127802411+dafeliton@users.noreply.github.com>
Date: Thu, 9 Jan 2025 10:23:40 -0800
Subject: [PATCH] Add cuda*

---
 images/scipy-ml-notebook/Dockerfile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
index ed4697e4..82fb4442 100644
--- 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
\ No newline at end of file
+RUN . /tmp/activate.sh