diff --git a/python/requirements/ml/py313/dl-cpu-requirements.txt b/python/requirements/ml/py313/dl-cpu-requirements.txt index f9ad5166f2f..e0639318586 100644 --- a/python/requirements/ml/py313/dl-cpu-requirements.txt +++ b/python/requirements/ml/py313/dl-cpu-requirements.txt @@ -1,7 +1,7 @@ # These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch. # For GPU Docker images, you should install dl-gpu-requirements.txt afterwards. -tensorflow==2.20.0; sys_platform != 'darwin' or platform_machine != 'arm64' +tensorflow==2.21.0; sys_platform != 'darwin' or platform_machine != 'arm64' tensorflow-macos==2.20.0; sys_platform == 'darwin' and platform_machine == 'arm64' tensorflow-probability==0.24.0 tensorflow-io-gcs-filesystem==0.31.0; python_version < '3.12' diff --git a/python/requirements/ml/py313/dl-gpu-requirements.txt b/python/requirements/ml/py313/dl-gpu-requirements.txt index e803faa2382..935308e2a08 100644 --- a/python/requirements/ml/py313/dl-gpu-requirements.txt +++ b/python/requirements/ml/py313/dl-gpu-requirements.txt @@ -1,6 +1,6 @@ # If you make changes below this line, please also make the corresponding changes to `dl-cpu-requirements.txt`! -tensorflow==2.20.0; sys_platform != 'darwin' or platform_machine != 'arm64' +tensorflow==2.21.0; sys_platform != 'darwin' or platform_machine != 'arm64' tensorflow-macos==2.20.0; sys_platform == 'darwin' and platform_machine == 'arm64' tensorflow-probability==0.24.0 tf-keras==2.20.0