Merge pull request #681 from IBM/romeokienzler-patch-3

GPU support in Dockerfile, trigger nightly build

Author: romeokienzler

.github/workflows/container_nightly.yaml

@@ -0,0 +1,33 @@
+name: Build & Push Docker Image
+
+on:
+  schedule:
+    - cron: '0 0 * * *'
+
+  workflow_dispatch:
+  
+jobs:
+  build-and-push:
+    if: github.ref == 'refs/heads/main'
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Log in to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build and push image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          push: true
+          tags: ghcr.io/ibm/terratorch:nightly

.github/workflows/container_stream.yaml

@@ -0,0 +1,32 @@
+name: Build & Push Docker Image (stream)
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  build-and-push:
+    if: github.ref == 'refs/heads/main'
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Log in to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build and push image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          push: true
+          tags: ghcr.io/ibm/terratorch:stream

Dockerfile

@@ -1,2 +1,32 @@
-FROM registry.access.redhat.com/ubi8/python-312:latest
-RUN pip install terratorch
+FROM nvidia/cuda:12.9.0-cudnn-runtime-ubuntu24.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN mkdir -p /opt/app-root/src
+RUN chown -R 10000:0 /opt/app-root
+RUN chmod -R g+rwX /opt/app-root
+
+RUN ln -fs /usr/share/zoneinfo/America/New_York /etc/localtime
+RUN  apt update && apt install -y tzdata 
+RUN dpkg-reconfigure --frontend noninteractive tzdata
+
+RUN apt-get update && apt-get install -y python3 python3-venv python3-pip git build-essential 
+RUN apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
+
+WORKDIR /opt/app-root/src
+
+# Create venv and install terratorch
+USER 10000
+RUN python3 -m venv /opt/app-root/src/venv
+RUN chgrp 0 -R /opt/app-root/src/venv/
+RUN chmod 775 -R /opt/app-root/src/venv
+RUN . /opt/app-root/src/venv/bin/activate && pip install  --no-cache-dir --upgrade pip && pip install  --no-cache-dir terratorch
+USER root
+RUN pip cache purge
+
+RUN apt-get purge -y build-essential && \
+    apt-get autoremove -y && \
+    rm -rf /var/lib/apt/lists/*
+
+
+CMD ["bash"]