bump gpu to torch 2.1

RasmusOrsoe · Oct 8, 2023 · 057f1be · 057f1be
1 parent fc93799
commit 057f1be
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
+torch==2.1.1+cu117
 --find-links https://data.pyg.org/whl/torch-2.1.0+cu117.html