env.sh 1.2 KB

123456789101112131415161718192021222324252627282930
  1. # Common environment setup across build*.sh scripts
  2. export VERSION=${VERSION:-$(git describe --tags --first-parent --abbrev=7 --long --dirty --always | sed -e "s/^v//g")}
  3. export GOFLAGS="'-ldflags=-w -s \"-X=github.com/ollama/ollama/version.Version=$VERSION\" \"-X=github.com/ollama/ollama/server.mode=release\"'"
  4. # TODO - consider `docker buildx ls --format=json` to autodiscover platform capability
  5. PLATFORM=${PLATFORM:-"linux/arm64,linux/amd64"}
  6. DOCKER_ORG=${DOCKER_ORG:-"ollama"}
  7. RELEASE_IMAGE_REPO=${RELEASE_IMAGE_REPO:-"${DOCKER_ORG}/release"}
  8. FINAL_IMAGE_REPO=${FINAL_IMAGE_REPO:-"${DOCKER_ORG}/ollama"}
  9. OLLAMA_COMMON_BUILD_ARGS="--build-arg=VERSION \
  10. --build-arg=GOFLAGS \
  11. --build-arg=OLLAMA_CUSTOM_CPU_DEFS \
  12. --build-arg=OLLAMA_SKIP_CUDA_GENERATE \
  13. --build-arg=OLLAMA_SKIP_CUDA_11_GENERATE \
  14. --build-arg=OLLAMA_SKIP_CUDA_12_GENERATE \
  15. --build-arg=CUDA_V11_ARCHITECTURES \
  16. --build-arg=CUDA_V12_ARCHITECTURES \
  17. --build-arg=OLLAMA_SKIP_ROCM_GENERATE \
  18. --build-arg=OLLAMA_FAST_BUILD \
  19. --build-arg=AMDGPU_TARGETS"
  20. OLLAMA_NEW_RUNNERS=${OLLAMA_NEW_RUNNERS:-""}
  21. if [ -n "${OLLAMA_NEW_RUNNERS}" ]; then
  22. DOCKERFILE_DIR="./llama/"
  23. else
  24. DOCKERFILE_DIR="./"
  25. fi
  26. echo "Building Ollama"
  27. echo "VERSION=$VERSION"
  28. echo "PLATFORM=$PLATFORM"