From 0b0508ec8f7a8c54649c765abeba39cf446107de Mon Sep 17 00:00:00 2001
From: Lars Reimann <mail@larsreimann.com>
Date: Sun, 5 May 2024 15:43:27 +0200
Subject: [PATCH] docs: install CUDA versions of torch

---
 docs/README.md | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docs/README.md b/docs/README.md
index 67e3ef6..67ff1de 100644
--- 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: