From dea146439ccd0a310c6d4154674e1fe7ed04108b Mon Sep 17 00:00:00 2001 From: Codersheepchen Date: Thu, 7 Nov 2024 19:33:59 +0800 Subject: [PATCH] keep runtime dependency consistent with compile --- tensorflow/tools/pip_package/setup.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index 992f2eae2..b4735adb2 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -58,10 +58,11 @@ REQUIRED_PACKAGES = [ 'google_pasta >= 0.1.6', 'keras_applications >= 1.0.8', 'keras_preprocessing >= 1.0.5', - 'numpy >= 1.16.0, < 2.0', + 'numpy == 1.18.5', + 'h5py == 2.10.0', 'opt_einsum >= 2.3.2', 'six >= 1.10.0', - 'protobuf >= 3.6.1', + 'protobuf == 3.19.0', 'tensorboard >= 1.15.0, < 1.16.0', 'tensorflow-estimator == 1.15.1', 'termcolor >= 1.1.0', -- Gitee