Update README.md to newer cuda version in readme install

openspyrit · Sep 13, 2024 · 090adf4 · 090adf4
1 parent 21db056
commit 090adf4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -34,7 +34,7 @@ pip3 install torch torchvision torchaudio
 (GPU version using `conda`)
 
 ``` shell
-conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
+conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
 ```
 
 Then, install SPyRiT using `pip`.