mirror of
https://github.com/TabbyML/tabby
synced 2024-11-21 16:03:07 +00:00
9da18587f5
When docker build then log show: ```txt 2 warnings found (use --debug to expand): - FromAsCasing: 'as' and 'FROM' keywords' casing do not match (line 9) - FromAsCasing: 'as' and 'FROM' keywords' casing do not match (line 49) ```
91 lines
2.6 KiB
Docker
91 lines
2.6 KiB
Docker
ARG UBUNTU_VERSION=22.04
|
|
# This needs to generally match the container host's environment.
|
|
ARG CUDA_VERSION=11.7.1
|
|
# Target the CUDA build image
|
|
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
|
# Target the CUDA runtime image
|
|
ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
|
|
|
|
FROM ${BASE_CUDA_DEV_CONTAINER} AS build
|
|
|
|
# Rust toolchain version
|
|
ARG RUST_TOOLCHAIN=stable
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
RUN apt-get update && \
|
|
apt-get install -y --no-install-recommends \
|
|
curl \
|
|
pkg-config \
|
|
libssl-dev \
|
|
protobuf-compiler \
|
|
git \
|
|
cmake \
|
|
&& \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# setup rust.
|
|
RUN curl https://sh.rustup.rs -sSf | bash -s -- --default-toolchain ${RUST_TOOLCHAIN} -y
|
|
ENV PATH="/root/.cargo/bin:${PATH}"
|
|
ENV LD_LIBRARY_PATH="/usr/local/cuda/compat/:${LD_LIBRARY_PATH}"
|
|
|
|
WORKDIR /root/workspace
|
|
|
|
RUN mkdir -p /opt/tabby/bin
|
|
RUN mkdir -p /opt/tabby/lib
|
|
RUN mkdir -p target
|
|
|
|
COPY . .
|
|
|
|
RUN --mount=type=cache,target=/usr/local/cargo/registry \
|
|
--mount=type=cache,target=/root/workspace/target \
|
|
cargo build --no-default-features --features cuda,prod --release --package tabby && \
|
|
cp target/release/llama-server /opt/tabby/bin/ && \
|
|
cp target/release/tabby /opt/tabby/bin/
|
|
|
|
# For compatibility with the legacy cpu build.
|
|
RUN cp /opt/tabby/bin/tabby /opt/tabby/bin/tabby-cpu
|
|
|
|
FROM ${BASE_CUDA_RUN_CONTAINER} AS runtime
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y --no-install-recommends \
|
|
git \
|
|
curl \
|
|
unzip \
|
|
openssh-client \
|
|
ca-certificates \
|
|
libgomp1 \
|
|
&& \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install katana
|
|
RUN curl -L https://github.com/projectdiscovery/katana/releases/download/v1.1.0/katana_1.1.0_linux_amd64.zip -o katana.zip \
|
|
&& unzip katana.zip katana \
|
|
&& mv katana /usr/bin/ \
|
|
&& rm katana.zip
|
|
|
|
# Disable safe directory in docker
|
|
# Context: https://github.com/git/git/commit/8959555cee7ec045958f9b6dd62e541affb7e7d9
|
|
RUN git config --system --add safe.directory "*"
|
|
|
|
# Automatic platform ARGs in the global scope
|
|
# https://docs.docker.com/engine/reference/builder/#automatic-platform-args-in-the-global-scope
|
|
ARG TARGETARCH
|
|
|
|
# AMD64 only:
|
|
# Make link to libnvidia-ml.so (NVML) library
|
|
# so that we could get GPU stats.
|
|
RUN if [ "$TARGETARCH" = "amd64" ]; then \
|
|
ln -s /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1 \
|
|
/usr/lib/x86_64-linux-gnu/libnvidia-ml.so; \
|
|
fi
|
|
|
|
COPY --from=build /opt/tabby /opt/tabby
|
|
|
|
ENV PATH="$PATH:/opt/tabby/bin"
|
|
ENV TABBY_ROOT=/data
|
|
|
|
ENTRYPOINT ["/opt/tabby/bin/tabby"]
|