Change Dockerfile

Dockerfile

@@ -4,16 +4,15 @@ FROM ghcr.io/huggingface/text-generation-inference:3.3.4
 # ---------- OS-level build tools & minimal CUDA compiler ----------
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-        build-essential git cmake ninja-build pkg-config curl ca-certificates \
-        libgl1-mesa-glx libegl1-mesa-dev libxrender-dev libxi-dev && \
-    curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub | \
-        gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg && \
+        build-essential git cmake ninja-build pkg-config curl ca-certificates gnupg && \
+    curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | \
+        gpg --dearmor --batch --yes -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/ubuntu2004/x86_64 /" \
+         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 --no-install-recommends cuda-compiler-11-8 && \
-    rm -rf /var/lib/apt/lists/*
+     apt-get update && \
+     apt-get install -y --no-install-recommends cuda-compiler-11-8 && \
+     rm -rf /var/lib/apt/lists/*
 
 # ---------- Make CMake happy if it looks for `cicc` ----------
 RUN if ! command -v cicc >/dev/null; then \