Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
-# Use the NVIDIA CUDA runtime as a base image for CUDA 11.8
-FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
@@ -47,18 +46,6 @@ RUN apt-get update && apt-get install -y \
     && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
     && apt-get clean
 
-# Install CUDA 11.8 and cuDNN 9.3.0
-RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | gpg --batch --yes --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg \
-    && echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /" > /etc/apt/sources.list.d/cuda.list \
-    && apt-get update && apt-get install -y \
-    libcudnn9=8.9.3.0-1+cuda11.8 \
-    libcudnn9-dev=8.9.3.0-1+cuda11.8 \
-    libcudnn9-doc=8.9.3.0-1+cuda11.8 \
-    && apt-get clean
-
-# Verify CUDA and cuDNN installation
-RUN dpkg-query -W libcudnn9
-
 # Switch back to the user
 USER user