diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 455f5f00f05..4fa078db880 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -5,7 +5,7 @@ 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.9.0 +torch==2.9.0+cpu torch-xla==2.8.1;sys_platform != 'darwin' # Jax cpu-only version (needed for testing). diff --git a/requirements.txt b/requirements.txt index 926a3ec883d..857da6af2a5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,13 +2,13 @@ # Note: when the version of Tensorflow is changed, the version tf_keras must be # changed in .github/workflows/actions.yml (pip install --no-deps tf_keras). tensorflow-cpu~=2.18.1;sys_platform != 'darwin' -tensorflow~=2.18.1;sys_platform == 'darwin' +tensorflow~=2.20.0;sys_platform == 'darwin' tf2onnx # Torch. --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 -torch-xla==2.6.0;sys_platform != 'darwin' +torch==2.9.0+cpu +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.