gpu: Fix NVIDIA CUDA example

coderefinery · Oct 11, 2024 · ddcceec · ddcceec
1 parent 201f05b
commit ddcceec
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/content/gpus.rst b/content/gpus.rst
@@ -22,11 +22,9 @@ When using NVIDIA's GPUs that use the CUDA-framework the flag is ``--nv``.
 
 As an example, let's get a CUDA-enabled PyTorch-image:
 
-:download:`accelerate_cuda.def </examples/accelerate_cuda.def>`:
-
-.. literalinclude:: /examples/accelerate-cuda.def
-   :language: singularity
+.. code-block:: console
 
+   $ apptainer pull pytorch-cuda.sif docker://docker.io/pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
 Now when we launch the image, we can give the image GPU access with