build_linux.sh 1.11 KB
Newer Older
Michael Yang's avatar
Michael Yang committed
1
#!/bin/sh
2

Michael Yang's avatar
Michael Yang committed
3
4
set -eu

5
export VERSION=${VERSION:-$(git describe --tags --first-parent --abbrev=7 --long --dirty --always | sed -e "s/^v//g")}
6
export GOFLAGS="'-ldflags=-w -s \"-X=github.com/ollama/ollama/version.Version=$VERSION\" \"-X=github.com/ollama/ollama/server.mode=release\"'"
7

8
BUILD_ARCH=${BUILD_ARCH:-"amd64 arm64"}
9
export AMDGPU_TARGETS=${AMDGPU_TARGETS:=""}
10
11
mkdir -p dist

12
for TARGETARCH in ${BUILD_ARCH}; do
13
14
15
16
17
    docker build \
        --platform=linux/$TARGETARCH \
        --build-arg=GOFLAGS \
        --build-arg=CGO_CFLAGS \
        --build-arg=OLLAMA_CUSTOM_CPU_DEFS \
18
        --build-arg=AMDGPU_TARGETS \
19
20
21
22
        --target build-$TARGETARCH \
        -f Dockerfile \
        -t builder:$TARGETARCH \
        .
23
    docker create --platform linux/$TARGETARCH --name builder-$TARGETARCH builder:$TARGETARCH
Daniel Hiltgen's avatar
Daniel Hiltgen committed
24
    docker cp builder-$TARGETARCH:/go/src/github.com/ollama/ollama/dist/linux-$TARGETARCH ./dist
25
    docker rm builder-$TARGETARCH
Daniel Hiltgen's avatar
Daniel Hiltgen committed
26
27
28
    echo "Compressing final linux bundle..."
    rm -f ./dist/ollama-linux-$TARGETARCH.tgz
    (cd dist/linux-$TARGETARCH && tar cf - . | gzip --best > ../ollama-linux-$TARGETARCH.tgz )
29
done