diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index 222e4400b..5332866e2 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.17 -tensorflow-text~=2.17 +tensorflow-cpu~=2.18 +tensorflow-text~=2.18 # Torch cpu-only version. --extra-index-url https://download.pytorch.org/whl/cpu diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index 20e77fadf..e9e9de499 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow with cuda support. -tensorflow[and-cuda]~=2.17 -tensorflow-text~=2.17 +tensorflow[and-cuda]~=2.18 +tensorflow-text~=2.18 # Torch cpu-only version. --extra-index-url https://download.pytorch.org/whl/cpu diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 10a49760a..2a601f0f2 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,11 +1,11 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.17 -tensorflow-text~=2.17 +tensorflow-cpu~=2.18 +tensorflow-text~=2.18 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.4.1+cu121 -torchvision==0.19.1+cu121 +torch==2.5.1+cu121 +torchvision==0.20.1+cu121 # Jax cpu-only version. jax[cpu] diff --git a/requirements.txt b/requirements.txt index 63ed7e90c..cc416f6ad 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # Tensorflow. -tensorflow-cpu~=2.17 -tensorflow-text~=2.17 +tensorflow-cpu~=2.18 +tensorflow-text~=2.18 # Torch. --extra-index-url https://download.pytorch.org/whl/cpu