From 07669114d6625285d1d3bf215f92306b66c9c996 Mon Sep 17 00:00:00 2001
From: Julio Perez <37191411+jperez999@users.noreply.github.com>
Date: Tue, 30 May 2023 14:11:28 -0400
Subject: [PATCH] revert tensorflow install (#997)

---
 docker/dockerfile.tf | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/dockerfile.tf b/docker/dockerfile.tf
index 6d80ba733..bff6f94af 100644
--- a/docker/dockerfile.tf
+++ b/docker/dockerfile.tf
@@ -17,8 +17,8 @@ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow2 back
 # Tensorflow dependencies (only)
 # Pinning to pass hugectr sok tests
 # wrapt 1.5.0 introduce hugectr test failures, so downgrade to 1.14.0
-RUN pip install --no-cache-dir tensorflow protobuf==3.20.3 wrapt==1.14.0 \
-    && pip uninstall tensorflow keras -y
+RUN pip install --no-cache-dir tensorflow-gpu==2.9.2 protobuf==3.20.3 wrapt==1.14.0 \
+    && pip uninstall tensorflow-gpu keras -y
 
 # DLFW Tensorflow packages
 COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python3.8/dist-packages/tensorflow /usr/local/lib/python3.8/dist-packages/tensorflow/