File size: 1,269 Bytes
283a882 | 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 | #!/bin/bash
# ============================================================================
# Build and push the ASCAD worker Docker image
# Run this on a machine with Docker access (e.g., Vast.ai instance)
# ============================================================================
# Usage: DOCKER_PAT=xxx bash build_docker.sh
# ============================================================================
set -euo pipefail
IMAGE_NAME="lemousehunter/ascad-worker:latest"
echo "=== Building Docker image ==="
cat > /tmp/Dockerfile <<'DOCKERFILE'
FROM tensorflow/tensorflow:2.16.2-gpu
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
wget \
unzip \
screen \
&& rm -rf /var/lib/apt/lists/*
RUN pip3 install --no-cache-dir \
h5py \
numpy \
scipy \
scikit-learn \
requests \
pyyaml \
wandb \
huggingface_hub
RUN mkdir -p /tmp/ascad_data /root/ascad-training-pipeline
WORKDIR /root
DOCKERFILE
docker build -t "$IMAGE_NAME" -f /tmp/Dockerfile /tmp
echo "=== Logging in to Docker Hub ==="
echo "${DOCKER_PAT}" | docker login -u lemousehunter --password-stdin
echo "=== Pushing image ==="
docker push "$IMAGE_NAME"
echo "=== Done! Image pushed: $IMAGE_NAME ==="
|