Add jupyter/pytorch-notebook

.github/ISSUE_TEMPLATE/bug_report.yml

@@ -26,6 +26,7 @@ body:
         - julia-notebook
         - minimal-notebook
         - pyspark-notebook
+        - pytorch-notebook
         - r-notebook
         - scipy-notebook
         - tensorflow-notebook

.github/ISSUE_TEMPLATE/feature_request.yml

@@ -24,6 +24,7 @@ body:
         - julia-notebook
         - minimal-notebook
         - pyspark-notebook
+        - pytorch-notebook
         - r-notebook
         - scipy-notebook
         - tensorflow-notebook

.github/actions/download-manifests/action.yml

@@ -83,6 +83,11 @@ runs:
       with:
         name: tensorflow-notebook-x86_64-history_line
         path: ${{ inputs.histLineDir }}
+    - name: Download artifact 📥
+      uses: actions/download-artifact@v3
+      with:
+        name: pytorch-notebook-x86_64-history_line
+        path: ${{ inputs.histLineDir }}
     - name: Download artifact 📥
       uses: actions/download-artifact@v3
       with:

.github/workflows/docker.yml

@@ -27,6 +27,7 @@ on:
       - "julia-notebook/**"
       - "minimal-notebook/**"
       - "pyspark-notebook/**"
+      - "pytorch-notebook/**"
       - "r-notebook/**"
       - "scipy-notebook/**"
       - "tensorflow-notebook/**"
@@ -55,6 +56,7 @@ on:
       - "julia-notebook/**"
       - "minimal-notebook/**"
       - "pyspark-notebook/**"
+      - "pytorch-notebook/**"
       - "r-notebook/**"
       - "scipy-notebook/**"
       - "tensorflow-notebook/**"
@@ -192,6 +194,15 @@ jobs:
       runsOn: ARM64
     needs: [aarch64-scipy]
 
+  aarch64-pytorch:
+    uses: ./.github/workflows/docker-build-test-upload.yml
+    with:
+      parentImage: scipy-notebook
+      image: pytorch-notebook
+      platform: aarch64
+      runsOn: ARM64
+    needs: [aarch64-scipy]
+
   x86_64-tensorflow:
     uses: ./.github/workflows/docker-build-test-upload.yml
     with:
@@ -201,6 +212,15 @@ jobs:
       runsOn: ubuntu-latest
     needs: [x86_64-scipy]
 
+  x86_64-pytorch:
+    uses: ./.github/workflows/docker-build-test-upload.yml
+    with:
+      parentImage: scipy-notebook
+      image: pytorch-notebook
+      platform: x86_64
+      runsOn: ubuntu-latest
+    needs: [x86_64-scipy]
+
   aarch64-datascience:
     uses: ./.github/workflows/docker-build-test-upload.yml
     with:
@@ -277,6 +297,7 @@ jobs:
             r-notebook,
             julia-notebook,
             tensorflow-notebook,
+            pytorch-notebook,
             datascience-notebook,
             pyspark-notebook,
             all-spark-notebook,
@@ -289,6 +310,7 @@ jobs:
         aarch64-scipy,
         aarch64-r,
         aarch64-tensorflow,
+        aarch64-pytorch,
         aarch64-datascience,
         aarch64-pyspark,
         aarch64-all-spark,
@@ -314,6 +336,7 @@ jobs:
             r-notebook,
             julia-notebook,
             tensorflow-notebook,
+            pytorch-notebook,
             datascience-notebook,
             pyspark-notebook,
             all-spark-notebook,
@@ -326,6 +349,7 @@ jobs:
         x86_64-scipy,
         x86_64-r,
         x86_64-tensorflow,
+        x86_64-pytorch,
         x86_64-datascience,
         x86_64-pyspark,
         x86_64-all-spark,
@@ -349,6 +373,7 @@ jobs:
             r-notebook,
             julia-notebook,
             tensorflow-notebook,
+            pytorch-notebook,
             datascience-notebook,
             pyspark-notebook,
             all-spark-notebook,

.github/workflows/hub-overview.yml

@@ -20,6 +20,7 @@ on:
       - "r-notebook/README.md"
       - "scipy-notebook/README.md"
       - "tensorflow-notebook/README.md"
+      - "pytorch-notebook/README.md"
 
 jobs:
   update-dockerhub-overview:
@@ -59,6 +60,8 @@ jobs:
             description: "Julia Jupyter Notebook Stack from https://github.com/jupyter/docker-stacks"
           - image: tensorflow-notebook
             description: "Scientific Jupyter Notebook Python Stack w/ TensorFlow from https://github.com/jupyter/docker-stacks"
+          - image: pytorch-notebook
+            description: "Scientific Jupyter Notebook Python Stack w/ PyTorch from https://github.com/jupyter/docker-stacks"
           - image: datascience-notebook
             description: "Data Science Jupyter Notebook Python Stack from https://github.com/jupyter/docker-stacks"
           - image: pyspark-notebook

Makefile

@@ -16,6 +16,7 @@ ALL_IMAGES:= \
 	julia-notebook \
 	scipy-notebook \
 	tensorflow-notebook \
+	pytorch-notebook \
 	datascience-notebook \
 	pyspark-notebook \
 	all-spark-notebook

docs/using/selecting.md

@@ -176,6 +176,17 @@ It contains:
 - Everything in `jupyter/scipy-notebook` and its ancestor images
 - [tensorflow](https://www.tensorflow.org/) machine learning library
 
+### jupyter/pytorch-notebook
+
+[Source on GitHub](https://github.com/jupyter/docker-stacks/tree/main/pytorch-notebook) |
+[Dockerfile commit history](https://github.com/jupyter/docker-stacks/commits/main/pytorch-notebook/Dockerfile) |
+[Docker Hub image tags](https://hub.docker.com/r/jupyter/pytorch-notebook/tags/)
+
+`jupyter/pytorch-notebook` includes popular Python deep learning libraries.
+
+- Everything in `jupyter/scipy-notebook` and its ancestor images
+- [pytorch](https://pytorch.org/) machine learning library
+
 ### jupyter/datascience-notebook
 
 [Source on GitHub](https://github.com/jupyter/docker-stacks/tree/main/datascience-notebook) |
@@ -226,7 +237,7 @@ The following diagram depicts the build dependency tree of the core images. (i.e
 Any given image inherits the complete content of all ancestor images pointing to it.
 
 [![Image inheritance
-diagram](../images/inherit.svg)](http://interactive.blockdiag.com/?compression=deflate&src=eJyFjkFuwkAMRfecwsqKLuYACMEJuqPLSshJHDAZ7GjGIwSIuzPTRaWJWmX7_vP_br12Y894gucKoKcBk7fjoGKRHwQ72Gwz18AkhsYqGU0aLCDbdpWjJrVJLH3L-vPrADe2c85ZDAJ5wkgfDbg99HmFgouG3RjdoEn6n7ZS_l9W7trc4ESNWtWxyBUoxpWFr-grac6KFzue7pVVk-I0RhI1DF5vv7z5W80vYqYkHS1Oh0XjkjzjwnPTPU4Yxsqas-Kh925uvt4imKoO)
+diagram](../images/inherit.svg)](http://interactive.blockdiag.com/?compression=deflate&src=eJyFz8FOwzAMgOH7nsLqCQ55ADTBE3CDIxJyU5eZZnaUOJoK2rsv4YCUSlOvv784yRjULxPjF_weACaasQT7nFUs8w_BMzwda9fEJIbGKjVFTZaQ7Xioo6GMRax8yMPr-xtc2E51zmKQKBBmehzAvcBUb6HksqFfspu1yPS3rS2_N2vnxrrBiRqNqkvDXWjizMJnDB3atuay57h2qi_NDEaSNc1BL_99uEPjapr8ac_Vr2CtJJ52n5h2xXcJjDufiGuOmJZObVtzGILbyusNkda3zw)
 
 ### Builds
 

mypy.ini

@@ -42,5 +42,8 @@ ignore_missing_imports = True
 [mypy-tensorflow.*]
 ignore_missing_imports = True
 
+[mypy-torch.*]
+ignore_missing_imports = True
+
 [mypy-urllib3.*]
 ignore_missing_imports = True

pytorch-notebook/.dockerignore

@@ -0,0 +1,2 @@
+# Documentation
+README.md

pytorch-notebook/Dockerfile

@@ -0,0 +1,20 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+ARG OWNER=jupyter
+ARG BASE_CONTAINER=$OWNER/scipy-notebook
+FROM $BASE_CONTAINER
+
+LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
+
+# Fix: https://github.com/hadolint/hadolint/wiki/DL4006
+# Fix: https://github.com/koalaman/shellcheck/wiki/SC3014
+SHELL ["/bin/bash", "-o", "pipefail", "-c"]
+
+# Install PyTorch with pip
+# hadolint ignore=DL3013
+RUN pip install --no-cache-dir --index-url 'https://download.pytorch.org/whl/cpu' \
+    'torch' \
+    'torchvision' \
+    'torchaudio'  && \
+    fix-permissions "${CONDA_DIR}" && \
+    fix-permissions "/home/${NB_USER}"

pytorch-notebook/README.md

@@ -0,0 +1,13 @@
+# Jupyter Notebook Deep Learning Stack
+
+[![docker pulls](https://img.shields.io/docker/pulls/jupyter/pytorch-notebook.svg)](https://hub.docker.com/r/jupyter/pytorch-notebook/)
+[![docker stars](https://img.shields.io/docker/stars/jupyter/pytorch-notebook.svg)](https://hub.docker.com/r/jupyter/pytorch-notebook/)
+[![image size](https://img.shields.io/docker/image-size/jupyter/pytorch-notebook/latest)](https://hub.docker.com/r/jupyter/pytorch-notebook/ "jupyter/pytorch-notebook image size")
+
+GitHub Actions in the <https://github.com/jupyter/docker-stacks> project builds and pushes this image to Docker Hub.
+
+Please visit the project documentation site for help to use and contribute to this image and others.
+
+- [Jupyter Docker Stacks on ReadTheDocs](https://jupyter-docker-stacks.readthedocs.io/en/latest/index.html)
+- [Selecting an Image :: Core Stacks :: jupyter/pytorch-notebook](https://jupyter-docker-stacks.readthedocs.io/en/latest/using/selecting.html#jupyter-pytorch-notebook)
+- [Image Specifics :: PyTorch](https://jupyter-docker-stacks.readthedocs.io/en/latest/using/specifics.html#pytorch)

tagging/images_hierarchy.py

@@ -21,6 +21,7 @@
     JupyterNotebookVersionTagger,
     PythonMajorMinorVersionTagger,
     PythonVersionTagger,
+    PytorchVersionTagger,
     RVersionTagger,
     SHATagger,
     SparkVersionTagger,
@@ -72,6 +73,9 @@ class ImageDescription:
     "tensorflow-notebook": ImageDescription(
         parent_image="scipy-notebook", taggers=[TensorflowVersionTagger()]
     ),
+    "pytorch-notebook": ImageDescription(
+        parent_image="scipy-notebook", taggers=[PytorchVersionTagger()]
+    ),
     "datascience-notebook": ImageDescription(
         parent_image="scipy-notebook",
         taggers=[RVersionTagger(), JuliaVersionTagger()],

tagging/taggers.py

@@ -113,6 +113,12 @@ def tag_value(container: Container) -> str:
         return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
 
 
+class PytorchVersionTagger(TaggerInterface):
+    @staticmethod
+    def tag_value(container: Container) -> str:
+        return "pytorch-" + _get_pip_package_version(container, "pytorch")
+
+
 class JuliaVersionTagger(TaggerInterface):
     @staticmethod
     def tag_value(container: Container) -> str:

tests/images_hierarchy.py

@@ -15,6 +15,7 @@
     "r-notebook": "minimal-notebook",
     "julia-notebook": "minimal-notebook",
     "tensorflow-notebook": "scipy-notebook",
+    "pytorch-notebook": "scipy-notebook",
     "datascience-notebook": "scipy-notebook",
     "pyspark-notebook": "scipy-notebook",
     "all-spark-notebook": "pyspark-notebook",

tests/pytorch-notebook/units/unit_pytorch.py

@@ -0,0 +1,5 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+import torch
+
+print(torch.tensor([[1.0, 4.0, 7.0], [4.0, 9.0, 11.0]]))