Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
-# Grab a fresh copy of the Python image
-FROM python:3.11-slim
+FROM ghcr.io/ggerganov/llama.cpp:full-cuda
 
 # Install build and runtime dependencies
 RUN apt-get update && \
@@ -10,9 +9,10 @@ RUN apt-get update && \
     pkg-config \
     curl
 
-RUN pip install -U pip setuptools wheel && \
-  CMAKE_ARGS="-DLLAMA_CUBLAS=ON" FORCE_CMAKE=1 pip install --verbose llama-cpp-python[server]
+RUN make LLAMA_CUBLAS=1
 
+RUN pip install -U pip setuptools wheel && \
+  pip install --verbose llama-cpp-python[server]
 
 # Download model
 RUN mkdir model && \