env.sh 1.1 KB

12345678910111213141516171819202122232425
  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. FINAL_IMAGE_REPO=${FINAL_IMAGE_REPO:-"${DOCKER_ORG}/ollama"}
  8. OLLAMA_COMMON_BUILD_ARGS="--build-arg=VERSION \
  9. --build-arg=GOFLAGS \
  10. --build-arg=OLLAMA_CUSTOM_CPU_DEFS \
  11. --build-arg=OLLAMA_SKIP_CUDA_GENERATE \
  12. --build-arg=OLLAMA_SKIP_CUDA_11_GENERATE \
  13. --build-arg=OLLAMA_SKIP_CUDA_12_GENERATE \
  14. --build-arg=CUDA_V11_ARCHITECTURES \
  15. --build-arg=CUDA_V12_ARCHITECTURES \
  16. --build-arg=OLLAMA_SKIP_ROCM_GENERATE \
  17. --build-arg=OLLAMA_FAST_BUILD \
  18. --build-arg=CUSTOM_CPU_FLAGS \
  19. --build-arg=GPU_RUNNER_CPU_FLAGS \
  20. --build-arg=AMDGPU_TARGETS"
  21. echo "Building Ollama"
  22. echo "VERSION=$VERSION"
  23. echo "PLATFORM=$PLATFORM"