Skip to content

Commit

Permalink
add raster vision notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
mhwasil committed Apr 29, 2024
1 parent e4e44a9 commit e79457a
Show file tree
Hide file tree
Showing 3 changed files with 40 additions and 0 deletions.
13 changes: 13 additions & 0 deletions .github/workflows/docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
21 changes: 21 additions & 0 deletions raster-vision-notebook/cuda12-pytorch-2.2.2/Dockerfile
Original file line number Diff line number Diff line change
@@ -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 <[email protected]>"

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
6 changes: 6 additions & 0 deletions raster-vision-notebook/cuda12-pytorch-2.2.2/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
rastervision
seaborn
geopandas
ExifRead
shapely
rasterio

0 comments on commit e79457a

Please sign in to comment.