Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# Use the NVIDIA CUDA runtime as a base image for CUDA 12.3
-FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
+# Use CUDA base image without CuDNN
+FROM nvidia/cuda:12.3.2-runtime-ubuntu22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
@@ -55,7 +55,6 @@ RUN apt-get update && apt-get install -y \
     && rm cudnn-local-repo-ubuntu2204-9.3.0_1.0-1_amd64.deb \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
-
 # Switch back to the user
 USER user