add docker
This commit is contained in:
Executable
+53
@@ -0,0 +1,53 @@
|
||||
#!/usr/bin/env bash
|
||||
# Сборка образов llama-cpp-turboquant (CUDA 12 / CUDA 13).
|
||||
# Контекст — клон TheTom/llama-cpp-turboquant; Dockerfile'ы — в этом репозитории.
|
||||
#
|
||||
# ./docker/build.sh # CUDA 12 (RTX 30xx)
|
||||
# ./docker/build.sh 13 # CUDA 13 / sm_120 (RTX 50xx)
|
||||
# ./docker/build.sh 12 13 # оба
|
||||
#
|
||||
# Переменные: LLAMA_CPP_SRC, DOCKER (docker|podman)
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||||
SRC="${LLAMA_CPP_SRC:-$(cd "$ROOT/../llama-cpp-turboquant" && pwd)}"
|
||||
DOCKER="${DOCKER:-docker}"
|
||||
|
||||
if [[ ! -f "$SRC/CMakeLists.txt" ]]; then
|
||||
echo "llama.cpp source not found at: $SRC" >&2
|
||||
echo "Set LLAMA_CPP_SRC to the TheTom/llama-cpp-turboquant checkout." >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
build_one() {
|
||||
local ver="$1"
|
||||
local df tag
|
||||
case "$ver" in
|
||||
12)
|
||||
df="$ROOT/docker/server-cuda12.Dockerfile"
|
||||
tag="llama-turboquant:server-cuda12"
|
||||
;;
|
||||
13)
|
||||
df="$ROOT/docker/server-cuda13.Dockerfile"
|
||||
tag="llama-turboquant:server-cuda13"
|
||||
;;
|
||||
*)
|
||||
echo "Unknown CUDA major: $ver (use 12 or 13)" >&2
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
echo "==> Building $tag"
|
||||
echo " Dockerfile: $df"
|
||||
echo " Context: $SRC"
|
||||
"$DOCKER" build -f "$df" -t "$tag" "$SRC"
|
||||
}
|
||||
|
||||
if [[ $# -eq 0 ]]; then
|
||||
set -- 12
|
||||
fi
|
||||
|
||||
for ver in "$@"; do
|
||||
build_one "$ver"
|
||||
done
|
||||
Reference in New Issue
Block a user