docs: install CUDA versions of torch

Safe-DS · May 5, 2024 · 0b0508e · 0b0508e
1 parent b55f44e
commit 0b0508e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docs/README.md b/docs/README.md
@@ -15,6 +15,13 @@ Get the latest version from [PyPI](https://pypi.org/project/safe-ds-runner):
 pip install safe-ds-runner
 ```
 
+On a Windows PC with an NVIDIA graphics card, you may also want to install the CUDA versions of `torch` and
+`torchvision`:
+
+```shell
+pip install --upgrade torch torchvision --index-url https://download.pytorch.org/whl/cu121
+```
+
 ## Usage
 
 Start the runner server: