bump gpu to 2.1

RasmusOrsoe · Oct 8, 2023 · fc93799 · fc93799
1 parent 4c47d0f
commit fc93799
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 recommended for functional performance
 --find-links https://download.pytorch.org/whl/torch_stable.html
 torch==2.0.1+cu117
---find-links https://data.pyg.org/whl/torch-2.0.0+cu117.html
+--find-links https://data.pyg.org/whl/torch-2.1.0+cu117.html