mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-11-11 21:39:52 +00:00
30 lines
725 B
Docker
30 lines
725 B
Docker
|
ARG UBUNTU_VERSION=jammy
|
||
|
|
|
||
|
|
FROM ubuntu:$UBUNTU_VERSION as build
|
||
|
|
|
||
|
|
# Install build tools
|
||
|
|
RUN apt update && apt install -y git build-essential cmake wget
|
||
|
|
|
||
|
|
# Install Vulkan SDK
|
||
|
|
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
|
||
|
|
|
||
|
|
# Build it
|
||
|
|
WORKDIR /app
|
||
|
|
COPY . .
|
||
|
|
RUN mkdir build && \
|
||
|
|
cd build && \
|
||
|
|
cmake .. -DLLAMA_VULKAN=1 && \
|
||
|
|
cmake --build . --config Release --target main
|
||
|
|
|
||
|
|
# Clean up
|
||
|
|
WORKDIR /
|
||
|
|
RUN cp /app/build/bin/main /main && \
|
||
|
|
rm -rf /app
|
||
|
|
|
||
|
|
ENV LC_ALL=C.utf8
|
||
|
|
|
||
|
|
ENTRYPOINT [ "/main" ]
|