diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index 2d9627eefb..9a4341f837 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -1,5 +1,5 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.19 +tensorflow-cpu~=2.20 tensorflow-text~=2.19 # Torch cpu-only version. diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index 94ab86d63f..f422bde67b 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -1,5 +1,5 @@ # Tensorflow with cuda support. -tensorflow[and-cuda]~=2.19 +tensorflow[and-cuda]~=2.20 tensorflow-text~=2.19 # Torch cpu-only version. diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index f30fdcb5a2..929b986032 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,11 +1,11 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.19 +tensorflow-cpu~=2.20 tensorflow-text~=2.19 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu126 -torch==2.7.1+cu126 -torchvision==0.22.1+cu126 +torch==2.8.0+cu126 +torchvision==0.23.0+cu126 # Jax cpu-only version. jax[cpu] diff --git a/requirements.txt b/requirements.txt index e499522558..2300272513 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # Tensorflow. -tensorflow-cpu~=2.19.0;sys_platform != 'darwin' -tensorflow~=2.19.0;sys_platform == 'darwin' +tensorflow-cpu~=2.20.0;sys_platform != 'darwin' +tensorflow~=2.20.0;sys_platform == 'darwin' tensorflow-text~=2.19;platform_system != 'Windows' # Torch.