diff --git a/requirements_pytorch.txt b/requirements_pytorch.txt index d6135c1..8e615a9 100644 --- a/requirements_pytorch.txt +++ b/requirements_pytorch.txt @@ -1 +1 @@ -torch>=1.8.0 +torch>=2.8.0 diff --git a/requirements_tests.txt b/requirements_tests.txt index 19a6ec1..723d757 100644 --- a/requirements_tests.txt +++ b/requirements_tests.txt @@ -8,4 +8,4 @@ dm-haiku==0.0.9 jax==0.4.7 jaxlib==0.4.7 tensorflow==2.13.0 -torch==1.13.1 +torch==2.8.0