Merge pull request graphnet-team#613 from RasmusOrsoe/update-gpu-requ…

…irements Update gpu requirements.txt
RasmusOrsoe · Oct 10, 2023 · 8cfec57 · 8cfec57
2 parents aa2a9b1 + 9e35a54
commit 8cfec57
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt
@@ -1,4 +1,4 @@
-# Contains packages recommended for functional performance
+# Contains packages requirements for GPU installation
 --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