bump cudnn for torch

ucsd-ets · Sep 29, 2024 · d57d031 · d57d031
1 parent d028519
commit d57d031
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -1,5 +1,5 @@
 ARG BASE_TAG=latest
-FROM ghcr.io/ucsd-ets/datascience-notebook:${BASE_TAG}
+FROM ghcr.io/ucsd-ets/datascience-notebook:2024.4-stable
 
 USER root
 
@@ -10,8 +10,7 @@ USER root
 # Python/Mamba deps
 ## Package versions
 ## tf 2.13 does not work with torch 2.2.1. Both require conflicting versions of typing-extensions
-## CUDA 11: ARG CUDA_VERSION=11.8 CUDNN_VERSION=8.7.0.84 \
-ARG CUDA_VERSION=12.1 CUDNN_VERSION=8.9.2.26 LIBNVINFER=7.2.2 LIBNVINFER_MAJOR_VERSION=7 \
+ARG CUDA_VERSION=12.1 CUDNN_VERSION=9.1.0.70 LIBNVINFER=7.2.2 LIBNVINFER_MAJOR_VERSION=7 \
   TENSORFLOW_VERSION=2.17.0 KERAS_VERSION=3.5.0 TENSORRT_VERSION=8.6.1 TORCH_VERSION=2.4.1 \
   PROTOBUF_VERSION=3.20.3