adjust gpu versions in requirements.txt

RasmusOrsoe · Oct 9, 2023 · 06313b2 · 06313b2
1 parent aa2a9b1
commit 06313b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt
@@ -1,4 +1,4 @@
 # Contains packages recommended for functional performance
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==2.1.1+cu117
---find-links https://data.pyg.org/whl/torch-2.1.0+cu117.html
+torch==2.1.0+cu118
+--find-links https://data.pyg.org/whl/torch-2.1.0+cu118.html