diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index 343026cf6ff..e21c1cb1c5b 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -3,7 +3,7 @@ tensorflow-cpu~=2.16.1 # Pin to TF 2.16 # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch>=2.1.0 +torch>=2.1.0, <2.3.0 torchvision>=0.16.0 # Jax with cuda support. diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index 69be284f766..f3b946ddcfe 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -3,7 +3,7 @@ tensorflow[and-cuda]~=2.16.1 # Pin to TF 2.16 # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch>=2.1.0 +torch>=2.1.0, <2.3.0 torchvision>=0.16.0 # Jax cpu-only version (needed for testing). diff --git a/requirements.txt b/requirements.txt index bd34860fe0c..c759c9d1815 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,8 +2,9 @@ tensorflow-cpu~=2.16.1 # Pin to TF 2.16 # Torch. +# TODO: Pin to < 2.3.0 (GitHub issue #19602) --extra-index-url https://download.pytorch.org/whl/cpu -torch>=2.1.0 +torch>=2.1.0, <2.3.0 torchvision>=0.16.0 # Jax.