Merge pull request graphnet-team#664 from kayla-leonard/main

Update torch version in gpu requirements to fix mis-match (2.2)
mobra7 · Feb 23, 2024 · 84d5dc2 · 84d5dc2
2 parents 9735f51 + eaee95f
commit 84d5dc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements/torch_gpu.txt b/requirements/torch_gpu.txt
@@ -1,4 +1,4 @@
 # Contains packages requirements for GPU installation
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==2.1.0+cu118
+torch==2.2.0+cu118
 --find-links https://data.pyg.org/whl/torch-2.2.0+cu118.html