Polysome/build_and_push_docker.sh at main · computationalpathologygroup/Polysome · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#!/bin/bash
# Manual Docker image building and publishing script
# This script builds and pushes Docker images to GitHub Container Registry

set -e

# Configuration
REGISTRY="ghcr.io"
IMAGE_NAME="computationalpathologygroup/polysome"
VERSION="${1:-latest}"  # Default to 'latest' if no version specified

echo "=== Polysome Docker Build & Push ==="
echo "Registry: ${REGISTRY}"
echo "Image: ${IMAGE_NAME}"
echo "Version: ${VERSION}"
echo ""

# Check if logged in to GitHub Container Registry
echo "Checking GitHub Container Registry login..."
if ! docker info 2>/dev/null | grep -q "${REGISTRY}"; then
    echo "Please login to GitHub Container Registry first:"
    echo "  docker login ${REGISTRY}"
    echo ""
    echo "Generate a Personal Access Token (classic) with 'write:packages' scope at:"
    echo "  https://github.com/settings/tokens"
    exit 1
fi

# Build GPU image
echo ""
echo "1. Building GPU Docker image..."
docker build -t ${REGISTRY}/${IMAGE_NAME}:${VERSION} -f Dockerfile .
echo "   ✓ GPU image built"

# Build ARM64 image
echo ""
echo "2. Building ARM64 Docker image..."
docker build -t ${REGISTRY}/${IMAGE_NAME}:${VERSION}-arm64 -f Dockerfile.arm64 --platform linux/arm64 .
echo "   ✓ ARM64 image built"

# Push images
echo ""
echo "3. Pushing images to registry..."
docker push ${REGISTRY}/${IMAGE_NAME}:${VERSION}
echo "   ✓ GPU image pushed"

docker push ${REGISTRY}/${IMAGE_NAME}:${VERSION}-arm64
echo "   ✓ ARM64 image pushed"

# Tag as latest if version is not latest
if [ "${VERSION}" != "latest" ]; then
    echo ""
    echo "4. Tagging as latest..."
    docker tag ${REGISTRY}/${IMAGE_NAME}:${VERSION} ${REGISTRY}/${IMAGE_NAME}:latest
    docker tag ${REGISTRY}/${IMAGE_NAME}:${VERSION}-arm64 ${REGISTRY}/${IMAGE_NAME}:latest-arm64

    docker push ${REGISTRY}/${IMAGE_NAME}:latest
    docker push ${REGISTRY}/${IMAGE_NAME}:latest-arm64
    echo "   ✓ Tagged and pushed as latest"
fi

echo ""
echo "✓ Docker images published successfully!"
echo ""
echo "Images published:"
echo "  ${REGISTRY}/${IMAGE_NAME}:${VERSION}"
echo "  ${REGISTRY}/${IMAGE_NAME}:${VERSION}-arm64"
if [ "${VERSION}" != "latest" ]; then
    echo "  ${REGISTRY}/${IMAGE_NAME}:latest"
    echo "  ${REGISTRY}/${IMAGE_NAME}:latest-arm64"
fi
echo ""
echo "Users can pull with:"
echo "  docker pull ${REGISTRY}/${IMAGE_NAME}:${VERSION}"
echo "  docker pull ${REGISTRY}/${IMAGE_NAME}:${VERSION}-arm64"