Update Dockerfile

Dockerfile

@@ -9,12 +9,13 @@ RUN apt update && apt install -y wget gnupg software-properties-common && \
     apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
     add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
 
-# Install dependencies
+# Install CUDA Toolkit and other dependencies
 RUN apt update && \
+    apt install -y cuda-toolkit-12-2 && \
     apt install --no-install-recommends -y build-essential python3 python3-pip curl git cmake zlib1g-dev libblas-dev && \
-    apt install -y libcublas-dev && \
     apt clean && rm -rf /var/lib/apt/lists/*
 
+
 WORKDIR /app
 
 # Download ggml and mmproj models from HuggingFace