File size: 1,844 Bytes
bfa102d |
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 |
#!/bin/bash
set -e
# Configuration
DOCKER_USERNAME="techdaskalos"
IMAGE_NAME="cybersecchatbot"
VERSION="${1:-gpu}"
FULL_IMAGE="$DOCKER_USERNAME/$IMAGE_NAME:$VERSION"
echo "🏗️ Building GPU Docker image: $FULL_IMAGE"
echo "================================"
# Build the image for GPU
docker buildx build --platform linux/amd64 -f Dockerfile.base.gpu -t "$FULL_IMAGE" .
echo ""
echo "✅ Build complete!"
echo ""
echo "🧪 Testing the image locally (requires NVIDIA GPU)..."
echo " Run: docker run --gpus all -p 8000:8000 $FULL_IMAGE"
echo " Then visit: http://localhost:8000/test"
echo ""
read -p "Would you like to test locally before pushing? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
echo "Starting local test server with GPU support..."
echo "Press Ctrl+C to stop when done testing"
docker run --gpus all -p 8000:8000 "$FULL_IMAGE"
fi
echo ""
read -p "Push to Docker Hub? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
echo "📤 Pushing to Docker Hub..."
# Check if logged in
if ! docker info | grep -q "Username: $DOCKER_USERNAME"; then
echo "Please login to Docker Hub:"
docker login
fi
docker push "$FULL_IMAGE"
echo ""
echo "✅ Successfully pushed: $FULL_IMAGE"
echo ""
echo "📝 Next steps:"
echo " 1. Update your HF Space Dockerfile to:"
echo " FROM $FULL_IMAGE"
echo ""
echo " 2. Update HF Space to use GPU hardware"
echo ""
echo " 3. Commit and push to HF Spaces:"
echo " cp Dockerfile.gpu Dockerfile"
echo " git add Dockerfile"
echo " git commit -m \"Switch to GPU-enabled image: $FULL_IMAGE\""
echo " git push"
echo ""
echo " Your HF Space will deploy with GPU acceleration!"
else
echo "Skipped push. Image is ready locally as: $FULL_IMAGE"
fi
|