Docker Images For Pangeo Jupyter Environment
The images defined in this repository capture reproducible computing environments used by Pangeo Cloud. They build on top of the Ubuntu operating system and include conda environments with a curated set of Python packages for geospatial analysis. While initially intended for Pangeo Cloud, they can be used outside of Pangeo infrastructure too!
More details can be found in our documentation.
Images are hosted on DockerHub and on Quay.io
Image | Description | Size | Pulls |
---|---|---|---|
base-image | Foundational Dockerfile for builds | ||
base-notebook | minimally functional image for pangeo hubs | ||
pangeo-notebook | base-notebook + core earth science analysis packages | ||
pytorch-notebook | pangeo-notebook + GPU-enabled pytorch | ||
ml-notebook | pangeo-notebook + GPU-enabled tensorflow2 |
Click on the image name in the table above for a current list of installed packages and versions
graph TD;
base-image-->base-notebook;
base-notebook-->pangeo-notebook;
pangeo-notebook-->pytorch-notebook;
pangeo-notebook-->ml-notebook;
click base-image "https://hub.docker.com/r/pangeo/base-image" "Open this in a new tab" _blank
click base-notebook "https://hub.docker.com/r/pangeo/base-notebook" "Open this in a new tab" _blank
click pangeo-notebook "https://hub.docker.com/r/pangeo/pangeo-notebook" "Open this in a new tab" _blank
click pytorch-notebook "https://hub.docker.com/r/pangeo/pytorch-notebook" "Open this in a new tab" _blank
click ml-notebook "https://hub.docker.com/r/pangeo/ml-notebook" "Open this in a new tab" _blank
If you want to use this image on an HPC system (including a GPU system), we recommend using Singularity. Please see the Singularity guide.
The primary use of these Docker images is running on Pangeo Cloud deployments with dask-gateway. Generally, the dask-gateway library version built into the image must match the dask-gateway version deployed in the cloud environment. The follow table keeps track of the first time a new dask-gateway version appears in a tagged image:
dask-gateway | Image tag |
---|---|
0.9 | 2020.11.06 |
0.8 | 2020.07.28 |
0.7 | 2020.04.22 |
ml-notebook
image now contains JAX and TensorFlow with XLA enabled. Due to licensing issues, conda-forge does not have ptxas
, but ptxas
is needed for XLA to work correctly. Should you like to use JAX and/or TensorFlow with XLA optimization, please install ptxas
on your own, for example, by conda install -c nvidia cuda-nvcc
. At the time of writing (October 2022), JAX throws a compilation error if the ptxas
version is higher than the driver version. There does not exist an easy solution for K80 GPUs, but in the case of T4 GPUs, you should install conda install -c nvidia cuda-nvcc==11.6.*
to be safe. Alternatively for any GPU, you could set an environment variable to resolve the error caused by JAX: XLA_FLAGS="--xla_gpu_force_compilation_parallelism=1"
. The aforementioned error will be removed (and likely turned into a warning) in a future version of JAX. See https://github.com/google/jax/issues/12776#issuecomment-1276649134
pangeo/forge
image, built for use with pangeo-forge. It is
no longer actively maintained or used, but you can still use the historical tags
if you wish.