From 17893d1ce4594b5116e5c7ee63023572b6f56d73 Mon Sep 17 00:00:00 2001 From: Joaquin Anton Date: Thu, 26 May 2022 10:16:29 +0200 Subject: [PATCH] Pin protobuf to <4 Signed-off-by: Joaquin Anton --- qa/setup_packages.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/qa/setup_packages.py b/qa/setup_packages.py index f0f64096aa2..9fa3c42fd59 100755 --- a/qa/setup_packages.py +++ b/qa/setup_packages.py @@ -435,13 +435,13 @@ def get_pyvers_name(self, url, cuda_version): "mxnet-cu{cuda_v}"), CudaPackage("tensorflow-gpu", { "100" : [ - PckgVer("1.15.5", python_max_ver="3.7"), - PckgVer("2.3.4", python_max_ver="3.8")], + PckgVer("1.15.5", python_max_ver="3.7", dependencies=["protobuf<4"]), + PckgVer("2.3.4", python_max_ver="3.8", dependencies=["protobuf<4"])], "110" : [ - PckgVer("1.15.5", python_max_ver="3.7"), - PckgVer("2.8.1", python_min_ver="3.7"), - PckgVer("2.9.0", python_min_ver="3.7"), - PckgVer("1.15.5+nv22.03", python_min_ver="3.8", python_max_ver="3.8", alias="nvidia-tensorflow")] + PckgVer("1.15.5", python_max_ver="3.7", dependencies=["protobuf<4"]), + PckgVer("2.8.1", python_min_ver="3.7", dependencies=["protobuf<4"]), + PckgVer("2.9.0", python_min_ver="3.7", dependencies=["protobuf<4"]), + PckgVer("1.15.5+nv22.03", python_min_ver="3.8", python_max_ver="3.8", alias="nvidia-tensorflow", dependencies=["protobuf<4"])] }), CudaPackageExtraIndex("torch", { "101" : ["1.8.0"],