From e79457ab5122a5725811ddf7811202778eefa2fe Mon Sep 17 00:00:00 2001
From: Mohammad Wasil <mwasil@outlook.co.id>
Date: Mon, 29 Apr 2024 13:24:55 +0200
Subject: [PATCH] add raster vision notebook

---
 .github/workflows/docker.yml                  | 13 ++++++++++++
 .../cuda12-pytorch-2.2.2/Dockerfile           | 21 +++++++++++++++++++
 .../cuda12-pytorch-2.2.2/requirements.txt     |  6 ++++++
 3 files changed, 40 insertions(+)
 create mode 100644 raster-vision-notebook/cuda12-pytorch-2.2.2/Dockerfile
 create mode 100644 raster-vision-notebook/cuda12-pytorch-2.2.2/requirements.txt

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 4b7cc89..c490baf 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -130,6 +130,19 @@ jobs:
       push: ${{ github.event_name == 'push' }}
       runs-on: ubuntu-latest
 
+  raster-vision-notebook-cuda12:
+    needs: [ml-notebook-cuda12]
+    secrets: inherit
+    uses: ./.github/workflows/docker-build-test-upload.yml
+    with:
+      parent-image: ml-notebook
+      parent-variant: cuda12-pytorch-2.2.2
+      image: raster-vision-notebook
+      variant: cuda12-pytorch-2.2.2
+      context: raster-vision-notebook/cuda12-pytorch-2.2.2/
+      push: ${{ github.event_name == 'push' }}
+      runs-on: ubuntu-latest
+
   # cuda12-ubuntu20.04 for ros noetic
   cuda12-ubuntu20_04:
     secrets: inherit
diff --git a/raster-vision-notebook/cuda12-pytorch-2.2.2/Dockerfile b/raster-vision-notebook/cuda12-pytorch-2.2.2/Dockerfile
new file mode 100644
index 0000000..c0d93de
--- /dev/null
+++ b/raster-vision-notebook/cuda12-pytorch-2.2.2/Dockerfile
@@ -0,0 +1,21 @@
+ARG BASE_IMAGE=quay.io/a2s-institute/ml-notebook:cuda12-pytorch-2.2.2
+
+FROM $BASE_IMAGE
+
+LABEL maintainer="Mohammad Wasil <mohammad.wasil@h-brs.de>"
+
+USER $NB_USER
+
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
+
+USER root
+
+# clean up
+RUN rm /tmp/requirements.txt && \
+    /opt/conda/bin/conda clean -afy && \
+    find /opt/conda/ -follow -type f -name '*.a' -delete && \
+    find /opt/conda/ -follow -type f -name '*.pyc' -delete && \
+    find /opt/conda/ -follow -type f -name '*.js.map' -delete
+
+USER $NB_USER
diff --git a/raster-vision-notebook/cuda12-pytorch-2.2.2/requirements.txt b/raster-vision-notebook/cuda12-pytorch-2.2.2/requirements.txt
new file mode 100644
index 0000000..6293a7b
--- /dev/null
+++ b/raster-vision-notebook/cuda12-pytorch-2.2.2/requirements.txt
@@ -0,0 +1,6 @@
+rastervision
+seaborn
+geopandas
+ExifRead
+shapely
+rasterio