Default pytorch split - TEST UNDER REVIEW, DO NOT MERGE

.gitlab-ci.yml

@@ -17,10 +17,12 @@ include:
 
 .images_matrix:
   - DOCKERFILE_RELPATH: "geolab-default"
+  - DOCKERFILE_RELPATH: "geolab-gpu"
   # - DOCKERFILE_RELPATH: "mspass_shortcourse"
   # - DOCKERFILE_RELPATH: "mt_shortcourse"
 
 variables:
   CONTAINER_REGISTRY_PLATFORM: "AWS-PUB"
   DOCKERFILE_RELPATH_IS_IMAGE_NAME: "true"
   GITLAB_HOSTED_RUNNER_SIZE: "saas-linux-medium-amd64"
+  USE_TIMESTAMP_VERSION: "true"

geolab-default/Dockerfile

@@ -1,4 +1,4 @@
-FROM pangeo/pytorch-notebook:2025.01.24
+FROM pangeo/base-notebook:2025.08.14
 
 USER root
 ENV DEBIAN_FRONTEND=noninteractive

geolab-default/environment.yml

@@ -10,9 +10,8 @@ dependencies:
   - pip
   - pip:
     - awswrangler
-    - cvxpy
     - dascore
-    - earthscope-sdk==1.2.0b0
+    - earthscope-sdk==1.2.0b1
     - earthscope-cli==1.0.1
     - earthscopestraintools
     - hypoinvpy
@@ -21,9 +20,8 @@ dependencies:
     - jupyterlab_jupyterbook_navigation
     - jupyter-resource-usage
     - obspy==1.4.1
+    - polars
     - pygmt==0.14.2
     - pynlloc
     - pyocto
     - pyrocko
-    - seisbench
-    - tensorflow

geolab-gpu/Dockerfile

@@ -0,0 +1,31 @@
+FROM pangeo/pytorch-notebook:2025.01.24
+
+USER root
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PATH ${NB_PYTHON_PREFIX}/bin:$PATH
+
+# Needed for apt-key to work, gcc compiler for 
+RUN apt-get update -qq --yes > /dev/null \
+    && apt-get install --yes -qq gnupg2 > /dev/null \
+    && apt-get install gcc --yes \
+    && apt-get install gfortran --yes \
+    && apt-get install g++ --yes \
+    && apt-get install make \
+    && apt-get install ftp --yes \
+    && apt-get clean
+
+USER ${NB_USER}
+
+COPY environment.yml /tmp/
+
+RUN mamba env update --name ${CONDA_ENV} -f /tmp/environment.yml \
+    && pip cache purge \
+    && mamba clean -afy
+RUN conda install awscli==2.27.0
+
+# Set up the start command 
+USER ${NB_USER}
+RUN chmod +x start \
+    && cp start /srv/start
+
+ENTRYPOINT ["/srv/start"]

geolab-gpu/environment.yml

@@ -0,0 +1,30 @@
+# Environment we need *on top* of base PANGEO stack
+# Get list of pangeo packages from https://github.com/pangeo-data/pangeo-docker-images/blob/master/pangeo-notebook/packages.txt
+# BUT REMEMBER TO PICK THE HASH THAT CORRESPONDS TO OUR BASE IMAGE
+channels:
+  - conda-forge
+
+dependencies:
+  # Packages required by ES
+  - tiledb
+  - pip
+  - pip:
+    - awswrangler
+    - cvxpy
+    - dascore
+    - earthscope-sdk==1.2.0b1
+    - earthscope-cli==1.0.1
+    - earthscopestraintools
+    - hypoinvpy
+    - gnssrefl
+    - jupyter_contrib_nbextensions
+    - jupyterlab_jupyterbook_navigation
+    - jupyter-resource-usage
+    - obspy==1.4.1
+    - polars[gpu] --extra-index-url=https://pypi.nvidia.com
+    - pygmt==0.14.2
+    - pynlloc
+    - pyocto
+    - pyrocko
+    - seisbench
+    - tensorflow

geolab-gpu/info.yml

@@ -0,0 +1,5 @@
+description: "Earthscope GeoLab notebook hub default image"
+aboutText:
+  file: ./README.md
+usageText:
+  file: ./USAGE.md