diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index f1ffb0f91933..f946904900b6 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.19.0 tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.7.1+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 f895f0224154..b48f53200e6a 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.19.0 tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.7.1+cpu # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 6a8596003069..f06c008b5be9 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,13 +1,13 @@ # Tensorflow cpu-only version (needed for testing). -tensorflow-cpu~=2.18.1 +tensorflow-cpu~=2.19.0 tf2onnx # Torch with cuda support. # - torch is pinned to a version that is compatible with torch-xla # - torch-xla is pinned to a version that supports GPU (2.6 doesn't) --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.6.0 -torch-xla==2.6.0;sys_platform != 'darwin' +torch==2.7.1+cpu +torch-xla==2.7.0;sys_platform != 'darwin' # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements.txt b/requirements.txt index 8d150a4e989e..8f69d1f5bf08 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.19.0;sys_platform != 'darwin' +tensorflow~=2.19.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.7.1+cpu;sys_platform != 'darwin' torch==2.6.0;sys_platform == 'darwin' -torch-xla==2.6.0;sys_platform != 'darwin' +torch-xla==2.7.0;sys_platform != 'darwin' # Jax. # Pinned to 0.5.0 on CPU. JAX 0.5.1 requires Tensorflow 2.19 for saved_model_test.