Update Dockerfile

Dockerfile

@@ -10,15 +10,6 @@ WORKDIR /app
 # Set the CUDA_HOME environment variable
 ENV CUDA_HOME=/usr/local/cuda
 
-# Install CUDA dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    cuda-command-line-tools-11-8 \
-    libcublas-11-8 \
-    libcublas-dev-11-8 \
-    libcudnn8=8.4.1.50-1+cuda11.8 \
-    libcudnn8-dev=8.4.1.50-1+cuda11.8 \
-    && apt-get clean && rm -rf /var/lib/apt/lists/*
-
 # Copy and install Python dependencies
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt