From 1f6ec92e9a9383e8cbd272eb4df97e07ff3817b0 Mon Sep 17 00:00:00 2001 From: Simon Li Date: Sat, 7 Sep 2024 14:47:58 +0100 Subject: [PATCH 1/2] Move `CUDA enabled variant` to lower down --- docs/using/selecting.md | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/docs/using/selecting.md b/docs/using/selecting.md index d4cd71dc54..b2261f5583 100644 --- a/docs/using/selecting.md +++ b/docs/using/selecting.md @@ -16,15 +16,6 @@ This section provides details about the first. The Jupyter team maintains a set of Docker image definitions in the GitHub repository. The following sections describe these images, including their contents, relationships, and versioning strategy. -### CUDA enabled variant - -We provide CUDA accelerated version of `pytorch-notebook` and `tensorflow-notebook` images. -Prepend a CUDA prefix (versioned prefix like `cuda12-` for `pytorch-notebook` or just `cuda-` for `tensorflow-notebook`) to the image tag -to allow PyTorch or TensorFlow operations to use compatible NVIDIA GPUs for accelerated computation. -Note: We only build `pytorch-notebook` for 2 last major versions of CUDA, `tensorflow-notebook` image only supports the latest CUDA version listed in the [officially tested build configurations](https://www.tensorflow.org/install/source#gpu). - -For example, you can use an image `quay.io/jupyter/pytorch-notebook:cuda12-python-3.11.8` or `quay.io/jupyter/tensorflow-notebook:cuda-latest` - ### jupyter/docker-stacks-foundation [Source on GitHub](https://github.com/jupyter/docker-stacks/tree/main/images/docker-stacks-foundation) | @@ -248,6 +239,15 @@ It contains: [ggplot2](https://ggplot2.tidyverse.org) packages +### CUDA enabled variant + +We provide CUDA accelerated version of `pytorch-notebook` and `tensorflow-notebook` images. +Prepend a CUDA prefix (versioned prefix like `cuda12-` for `pytorch-notebook` or just `cuda-` for `tensorflow-notebook`) to the image tag +to allow PyTorch or TensorFlow operations to use compatible NVIDIA GPUs for accelerated computation. +Note: We only build `pytorch-notebook` for 2 last major versions of CUDA, `tensorflow-notebook` image only supports the latest CUDA version listed in the [officially tested build configurations](https://www.tensorflow.org/install/source#gpu). + +For example, you can use an image `quay.io/jupyter/pytorch-notebook:cuda12-python-3.11.8` or `quay.io/jupyter/tensorflow-notebook:cuda-latest` + ### Image Relationships The following diagram depicts the build dependency tree of the core images. (i.e., the `FROM` statements in their Dockerfiles). From e3df4c1883273f3466f900f049856cd63c0dc27e Mon Sep 17 00:00:00 2001 From: Simon Li Date: Sat, 7 Sep 2024 14:51:23 +0100 Subject: [PATCH 2/2] CUDA enabled variant: fix grammar --- docs/using/selecting.md | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/docs/using/selecting.md b/docs/using/selecting.md index b2261f5583..a3c77c7b4a 100644 --- a/docs/using/selecting.md +++ b/docs/using/selecting.md @@ -239,14 +239,15 @@ It contains: [ggplot2](https://ggplot2.tidyverse.org) packages -### CUDA enabled variant +### CUDA enabled variants -We provide CUDA accelerated version of `pytorch-notebook` and `tensorflow-notebook` images. +We provide CUDA accelerated versions of the `pytorch-notebook` and `tensorflow-notebook` images. Prepend a CUDA prefix (versioned prefix like `cuda12-` for `pytorch-notebook` or just `cuda-` for `tensorflow-notebook`) to the image tag to allow PyTorch or TensorFlow operations to use compatible NVIDIA GPUs for accelerated computation. -Note: We only build `pytorch-notebook` for 2 last major versions of CUDA, `tensorflow-notebook` image only supports the latest CUDA version listed in the [officially tested build configurations](https://www.tensorflow.org/install/source#gpu). +We only build `pytorch-notebook` for last two major versions of CUDA. +The `tensorflow-notebook` image only supports the latest CUDA version listed in the [officially tested build configurations](https://www.tensorflow.org/install/source#gpu). -For example, you can use an image `quay.io/jupyter/pytorch-notebook:cuda12-python-3.11.8` or `quay.io/jupyter/tensorflow-notebook:cuda-latest` +For example, you could use the image `quay.io/jupyter/pytorch-notebook:cuda12-python-3.11.8` or `quay.io/jupyter/tensorflow-notebook:cuda-latest`. ### Image Relationships