livebook/docker/build_and_push.sh
2023-10-09 21:43:50 +02:00

57 lines
1.6 KiB
Bash
Executable file

#!/bin/bash
# Builds extra Livebook images.
#
# Images are pushed to GHCR, so you need to make sure that the docker
# client is authenticated.
set -ex
cd "$(dirname "$0")/.."
. versions
livebook_version="$(cat mix.exs | grep '@version "' | grep -v '\-dev' | grep -oE '[0-9]{1,}.[0-9]{1,}.[0-9]{1,}' || true)"
if [ -z "$livebook_version" ]; then
echo "No stable Livebook version detected"
exit 1
fi
cuda_tag_list=("cuda11.8" "cuda12.1")
cuda_version_list=("11.8.0" "12.1.0")
for idx in "${!cuda_tag_list[@]}"; do
cuda_tag="${cuda_tag_list[idx]}"
cuda_version="${cuda_version_list[idx]}"
base_image="ghcr.io/livebook-dev/utils:elixir-$elixir-erlang-$otp-$cuda_tag"
if docker manifest inspect $base_image > /dev/null; then
echo "Using base image: $base_image"
else
echo "Building base image: $base_image"
docker buildx build --push --platform linux/amd64,linux/arm64 \
-t $base_image \
--build-arg ELIXIR_VERSION=$elixir \
--build-arg ERLANG_VERSION=$otp \
--build-arg UBUNTU_VERSION=$ubuntu \
--build-arg CUDA_VERSION=$cuda_version \
-f docker/base/elixir-cuda.dockerfile \
docker/base
fi
image="ghcr.io/livebook-dev/livebook:$livebook_version-$cuda_tag"
if docker manifest inspect $image > /dev/null; then
echo "Skipping image, since it already exists: $image"
else
echo "Building image: $image"
docker buildx build --push --platform linux/amd64,linux/arm64 \
--build-arg BASE_IMAGE="$base_image" \
-t $image \
-t ghcr.io/livebook-dev/livebook:latest-$cuda_tag \
.
fi
done