Trevor Starick
be936d6b59
- Introduce ARG REPO and ARG BRANCH to replace the hardcoded git clone with: `git clone -b ${BRANCH} --single-branch --recursive ${REPO}` . This allows overriding the llama.cpp repository and branch at build time via `--build-arg`.
- Update `docs/building.md` to recommend using `--build-arg` instead of updating the file
113 lines
3.1 KiB
Docker
113 lines
3.1 KiB
Docker
# build stage
|
|
FROM registry.fedoraproject.org/fedora:43 AS builder
|
|
|
|
# rocm 7.2 repo
|
|
RUN <<'EOF'
|
|
tee /etc/yum.repos.d/rocm.repo <<REPO
|
|
[ROCm-7.2]
|
|
name=ROCm7.2
|
|
baseurl=https://repo.radeon.com/rocm/rhel10/7.2/main
|
|
enabled=1
|
|
priority=50
|
|
gpgcheck=1
|
|
gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key
|
|
REPO
|
|
EOF
|
|
|
|
# deps
|
|
RUN dnf -y --nodocs --setopt=install_weak_deps=False \
|
|
--exclude='*sdk*' --exclude='*samples*' --exclude='*-doc*' --exclude='*-docs*' \
|
|
install \
|
|
make gcc cmake lld clang clang-devel compiler-rt libcurl-devel ninja-build \
|
|
rocm-llvm rocm-device-libs hip-runtime-amd hip-devel \
|
|
rocblas rocblas-devel hipblas hipblas-devel rocm-cmake libomp-devel libomp \
|
|
rocminfo radeontop \
|
|
git-core vim sudo rsync \
|
|
&& dnf clean all && rm -rf /var/cache/dnf/*
|
|
|
|
# rocm env
|
|
ENV ROCM_PATH=/opt/rocm \
|
|
HIP_PATH=/opt/rocm \
|
|
HIP_CLANG_PATH=/opt/rocm/llvm/bin \
|
|
HIP_DEVICE_LIB_PATH=/opt/rocm/amdgcn/bitcode \
|
|
PATH=/opt/rocm/bin:/opt/rocm/llvm/bin:$PATH
|
|
|
|
# llama.cpp
|
|
WORKDIR /opt/llama.cpp
|
|
ARG REPO=https://github.com/ggerganov/llama.cpp.git
|
|
ARG BRANCH=master
|
|
RUN git clone -b ${BRANCH} --single-branch --recursive ${REPO} . \
|
|
|
|
# build
|
|
RUN git clean -xdf \
|
|
&& git submodule update --recursive \
|
|
&& cmake -S . -B build \
|
|
-DGGML_HIP=ON \
|
|
-DCMAKE_HIP_FLAGS="--rocm-path=/opt/rocm -mllvm --amdgpu-unroll-threshold-local=600" \
|
|
-DAMDGPU_TARGETS=gfx1151 \
|
|
-DCMAKE_BUILD_TYPE=Release \
|
|
-DGGML_RPC=ON \
|
|
-DLLAMA_HIP_UMA=ON \
|
|
-DGGML_CUDA_ENABLE_UNIFIED_MEMORY=ON \
|
|
-DROCM_PATH=/opt/rocm \
|
|
-DHIP_PATH=/opt/rocm \
|
|
-DHIP_PLATFORM=amd \
|
|
&& cmake --build build --config Release -- -j$(nproc) \
|
|
&& cmake --install build --config Release
|
|
|
|
# libs
|
|
RUN find /opt/llama.cpp/build -type f -name 'lib*.so*' -exec cp {} /usr/lib64/ \; \
|
|
&& ldconfig
|
|
|
|
# helper
|
|
COPY gguf-vram-estimator.py /usr/local/bin/gguf-vram-estimator.py
|
|
RUN chmod +x /usr/local/bin/gguf-vram-estimator.py
|
|
|
|
# runtime stage
|
|
FROM registry.fedoraproject.org/fedora-minimal:43
|
|
|
|
# rocm 7.2 repo
|
|
RUN <<'EOF'
|
|
tee /etc/yum.repos.d/rocm.repo <<REPO
|
|
[ROCm-7.2]
|
|
name=ROCm7.2
|
|
baseurl=https://repo.radeon.com/rocm/rhel10/7.2/main
|
|
enabled=1
|
|
priority=50
|
|
gpgcheck=1
|
|
gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key
|
|
REPO
|
|
EOF
|
|
|
|
# runtime deps
|
|
RUN microdnf -y --nodocs --setopt=install_weak_deps=0 \
|
|
--exclude='*sdk*' --exclude='*samples*' --exclude='*-doc*' --exclude='*-docs*' \
|
|
install \
|
|
bash ca-certificates libatomic libstdc++ libgcc libgomp sudo \
|
|
hip-runtime-amd rocblas hipblas \
|
|
rocminfo radeontop procps-ng \
|
|
&& microdnf clean all && rm -rf /var/cache/dnf/*
|
|
|
|
# copy
|
|
COPY --from=builder /usr/local/ /usr/local/
|
|
COPY --from=builder /opt/llama.cpp/build/bin/rpc-* /usr/local/bin/
|
|
|
|
# ld
|
|
RUN echo "/usr/local/lib" > /etc/ld.so.conf.d/local.conf \
|
|
&& echo "/usr/local/lib64" >> /etc/ld.so.conf.d/local.conf \
|
|
&& ldconfig \
|
|
&& cp -n /usr/local/lib/libllama*.so* /usr/lib64/ 2>/dev/null || true \
|
|
&& ldconfig
|
|
|
|
# helper
|
|
COPY gguf-vram-estimator.py /usr/local/bin/gguf-vram-estimator.py
|
|
RUN chmod +x /usr/local/bin/gguf-vram-estimator.py
|
|
|
|
# profile
|
|
RUN printf '%s\n' \
|
|
> /etc/profile.d/rocm.sh && chmod +x /etc/profile.d/rocm.sh \
|
|
&& echo 'source /etc/profile.d/rocm.sh' >> /etc/bashrc
|
|
|
|
# shell
|
|
CMD ["/bin/bash"]
|