deps: Bump minimum pytorch version to 1.9.0

1.8.0 doesn't support 2d tensors in CrossEntropyLoss. Signed-off-by: Jan Vesely <[email protected]>
jvesely · Oct 19, 2023 · 2746e93 · 2746e93
1 parent 7898c6a
commit 2746e93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -19,4 +19,4 @@ pint<0.22.0
 protobuf<3.20.4
 rich>=10.1, <10.13
 toposort<1.11
-torch>=1.8.0, <2.1.0; (platform_machine == 'AMD64' or platform_machine == 'x86_64') and platform_python_implementation == 'CPython' and implementation_name == 'cpython'
+torch>=1.9.0, <2.1.0; (platform_machine == 'AMD64' or platform_machine == 'x86_64') and platform_python_implementation == 'CPython' and implementation_name == 'cpython'