diff --git a/doc/SETUP.md b/doc/SETUP.md
index e34acd1..b9c3dbd 100644
--- a/doc/SETUP.md
+++ b/doc/SETUP.md
@@ -48,6 +48,8 @@ Double check your python version again after setup by running these two commands
 
 Again, this should show 3.10.x
 
+Finally, install CUDA 11.8 from this link: `https://developer.nvidia.com/cuda-11-8-0-download-archive`
+
 ## Local docker container
 
 ```sh
@@ -95,4 +97,4 @@ Conda should be something like this (untested)
 * `cd EveryDream2trainer`
 * `conda create --name ed2 python=3.10`
 * `conda activate ed2`
-* `pip install -r requirements.txt`
\ No newline at end of file
+* `pip install -r requirements.txt`