Skip to content

Commit

Permalink
docs: Core Stacks: move CUDA docs to later in the page (jupyter#2145)
Browse files Browse the repository at this point in the history
* Move `CUDA enabled variant` to lower down

* CUDA enabled variant: fix grammar
  • Loading branch information
manics authored Sep 14, 2024
1 parent 1fb08d7 commit 8f9b1b6
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions docs/using/selecting.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,6 @@ This section provides details about the first.
The Jupyter team maintains a set of Docker image definitions in the <https://github.com/jupyter/docker-stacks> 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) |
Expand Down Expand Up @@ -248,6 +239,16 @@ It contains:
[ggplot2](https://ggplot2.tidyverse.org)
packages

### CUDA enabled variants

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.
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 could use the 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).
Expand Down

0 comments on commit 8f9b1b6

Please sign in to comment.