Skip to content

Commit 0e3c933

Browse files
rtg0795tensorflower-gardener
authored andcommitted
Remove pinning protobuf version since 3.20.3 is old and not supported by Tensorflow
PiperOrigin-RevId: 728286886
1 parent eaa4052 commit 0e3c933

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

WORKSPACE

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ bind(
3131

3232
http_archive(
3333
name = "com_google_protobuf",
34-
sha256 = "f66073dee0bc159157b0bd7f502d7d1ee0bc76b3c1eac9836927511bdc4b3fc1",
35-
strip_prefix = "protobuf-3.21.9",
36-
urls = ["https://github.com/protocolbuffers/protobuf/archive/v3.21.9.zip"],
34+
sha256 = "6b1040a5661cd5f6e610cbca9cfaa2a17d60e2bb545309bc1b278bb05be44bdd",
35+
strip_prefix = "protobuf-4.21.6",
36+
urls = ["https://files.pythonhosted.org/packages/0e/25/693cd589793e7ae429ef76ab08f74b7866d342fd079cc4c723141a9660d3/protobuf-4.21.6.tar.gz"],
3737
)
3838

3939
# ZLIB. Need by com_google_protobuf.

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ scipy
88
# with both new and old protobuf stubs. This is needed to resolve
99
# Keras-TensorFlow circular dependency issue, when one of them gets a dependency
1010
# incompatible with another one (protobuf in this specific case).
11-
protobuf==3.20.3
12-
tf-nightly
11+
protobuf==4.21.6
12+
tf-nightly[and_cuda]
1313
portpicker
1414
pyyaml
1515
Pillow

tf_keras/kokoro/github/ubuntu/gpu/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ pip uninstall -y keras-nightly
4242

4343
# LD Library Path needs to be same as TensorFlow Ubuntu Docker build -
4444
# https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/tf_sig_build_dockerfiles/
45-
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/tensorrt/lib"
45+
# export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/tensorrt/lib"
4646
CUDA_TOOLKIT_PATH="/usr/local/cuda-11.8"
4747
TF_CUDA_CONFIG_REPO="@ubuntu20.04-gcc9_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda"
4848
TF_CUDA_COMPUTE_CAPABILITIES="sm_35,sm_50,sm_60,sm_70,sm_75,compute_80"

0 commit comments

Comments
 (0)