summaryrefslogtreecommitdiff
path: root/.devops/llama-server-vulkan.Dockerfile
diff options
context:
space:
mode:
Diffstat (limited to '.devops/llama-server-vulkan.Dockerfile')
-rw-r--r--.devops/llama-server-vulkan.Dockerfile14
1 files changed, 6 insertions, 8 deletions
diff --git a/.devops/llama-server-vulkan.Dockerfile b/.devops/llama-server-vulkan.Dockerfile
index b0fa0b8e..13a61ffd 100644
--- a/.devops/llama-server-vulkan.Dockerfile
+++ b/.devops/llama-server-vulkan.Dockerfile
@@ -1,24 +1,20 @@
ARG UBUNTU_VERSION=jammy
-FROM ubuntu:$UBUNTU_VERSION as build
+FROM ubuntu:$UBUNTU_VERSION AS build
# Install build tools
RUN apt update && apt install -y git build-essential cmake wget
-# Install Vulkan SDK
+# Install Vulkan SDK and cURL
RUN wget -qO - https://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add - && \
wget -qO /etc/apt/sources.list.d/lunarg-vulkan-jammy.list https://packages.lunarg.com/vulkan/lunarg-vulkan-jammy.list && \
apt update -y && \
- apt-get install -y vulkan-sdk
-
-# Install cURL
-RUN apt-get update && \
- apt-get install -y libcurl4-openssl-dev
+ apt-get install -y vulkan-sdk libcurl4-openssl-dev curl
# Build it
WORKDIR /app
COPY . .
-RUN cmake -B build -DLLAMA_VULKAN=1 -DLLAMA_CURL=1 && \
+RUN cmake -B build -DGGML_VULKAN=1 -DLLAMA_CURL=1 && \
cmake --build build --config Release --target llama-server
# Clean up
@@ -28,4 +24,6 @@ RUN cp /app/build/bin/llama-server /llama-server && \
ENV LC_ALL=C.utf8
+HEALTHCHECK CMD [ "curl", "-f", "http://localhost:8080/health" ]
+
ENTRYPOINT [ "/llama-server" ]