From 2746e93fff9972a9d7e96e17cfa525ed6f23df20 Mon Sep 17 00:00:00 2001 From: Jan Vesely Date: Thu, 19 Oct 2023 12:32:58 -0400 Subject: [PATCH] deps: Bump minimum pytorch version to 1.9.0 1.8.0 doesn't support 2d tensors in CrossEntropyLoss. Signed-off-by: Jan Vesely --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 84fa3fae0dd..28616fbfb5a 100644 --- 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'