diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index f1ffb0f9193..c81c9600e83 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -1,10 +1,10 @@ # Tensorflow cpu-only version (needed for testing). -tensorflow-cpu~=2.18.1 +tensorflow-cpu~=2.20.0 tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.8.0+cpu # Jax with cuda support. --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index f895f022415..b345a2b2892 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -1,10 +1,10 @@ # Tensorflow with cuda support. -tensorflow[and-cuda]~=2.18.1 +tensorflow[and-cuda]~=2.20.0 tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.8.0+cpu # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 9bad75fb290..d971152e6e1 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,12 +1,12 @@ # Tensorflow cpu-only version (needed for testing). -tensorflow-cpu~=2.18.1 +tensorflow-cpu~=2.20.0 tf2onnx # Torch with cuda support. # - torch is pinned to a version that is compatible with torch-xla. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.6.0 -torch-xla==2.6.0;sys_platform != 'darwin' +torch==2.8.0+cpu +torch-xla==2.8.1;sys_platform != 'darwin' # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements.txt b/requirements.txt index e5a44501e6b..9ff1d46475a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,14 +1,14 @@ # Tensorflow. -tensorflow-cpu~=2.18.1;sys_platform != 'darwin' -tensorflow~=2.18.1;sys_platform == 'darwin' +tensorflow-cpu~=2.20.0;sys_platform != 'darwin' +tensorflow~=2.20.0;sys_platform == 'darwin' tf_keras tf2onnx # Torch. --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0;sys_platform != 'darwin' +torch==2.8.0+cpu;sys_platform != 'darwin' torch==2.6.0;sys_platform == 'darwin' -torch-xla==2.6.0;sys_platform != 'darwin' +torch-xla==2.8.1;sys_platform != 'darwin' # Jax. # Pinned to 0.5.0 on CPU. JAX 0.5.1 requires Tensorflow 2.19 for saved_model_test.