hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-18 07:22:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve CUDA enabled images docs

Browse Source
This commit is contained in:
Ayaz Salikhov
2024-04-03 01:36:57 +01:00
parent 275ad774c2
commit e838ff397a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/using/selecting.md
View File

@@ -19,7 +19,7 @@ The following sections describe these images, including their contents, relation
## CUDA enabled variant ## CUDA enabled variant
We provide CUDA accelerated version of `pytorch-notebook` and `tensorflow-notebook` images. We provide CUDA accelerated version of `pytorch-notebook` and `tensorflow-notebook` images.
Prepend a CUDA version prefix (like `cuda12-` for `pytorch-notebook` or `cuda-` for `tensorflow-notebook`) to the image tag 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. 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). 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).