From 1a791bcff2d597dd8dcfd2313027aaa783899e45 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B5=81=E5=B9=B4?= <williamdeyang@outlook.com>
Date: Fri, 6 Dec 2024 03:48:58 +0800
Subject: [PATCH] Update llama-server-cuda.Dockerfile

---
 .devops/llama-server-cuda.Dockerfile | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/.devops/llama-server-cuda.Dockerfile b/.devops/llama-server-cuda.Dockerfile
index 130bbd633..f5916f9d3 100644
--- a/.devops/llama-server-cuda.Dockerfile
+++ b/.devops/llama-server-cuda.Dockerfile
@@ -13,10 +13,11 @@ ENV TZ=Africa/Johannesburg
 # CUDA architecture to build for (defaults to all supported archs)
 ARG CUDA_DOCKER_ARCH=default
 
-RUN apt-get update && \
-    apt-get install -y build-essential git cmake libcurl4-openssl-dev
-
-WORKDIR /app
+RUN wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | sudo apt-key add -WORKDIR /app && \
+    apt-add-repository 'deb https://apt.kitware.com/ubuntu/ bionic main' && \
+    apt-get update && \
+    apt-get install -y build-essential git cmake libcurl4-openssl-dev && \
+    apt-get install cmake
 
 COPY . .