Skip to content

Commit cbfa7b8

Browse files
authored
Fix sok test error due to conflict version of tensorflow-metadata (#1057)
* Fix sok test error due to conflict version of tensorflow-metadata * Add missed python path for hugectr
1 parent cc863d7 commit cbfa7b8

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

docker/dockerfile.ctr

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ RUN if [[ "${HUGECTR_DEV_MODE}" == "false" ]]; then \
7777
mv /hugectr/ci ~/hugectr-ci && rm -rf /hugectr && mkdir -p /hugectr && mv ~/hugectr-ci /hugectr/ci \
7878
; fi
7979

80+
ENV PYTHONPATH=${PYTHONPATH}:${HUGECTR_HOME}/lib
8081
ARG _HUGECTR_BACKEND_REPO="github.com/triton-inference-server/hugectr_backend.git"
8182
ARG TRITON_VERSION
8283
# Install Triton inference backend.

docker/dockerfile.tf

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow backe
1717
# Tensorflow dependencies (only)
1818
# Pinning to pass hugectr sok tests
1919
# wrapt 1.5.0 introduce hugectr test failures, so downgrade to 1.14.0
20-
RUN pip install --no-cache-dir tensorflow protobuf==3.20.3 wrapt==1.14.0 \
21-
&& pip uninstall tensorflow keras -y
20+
RUN pip install --no-cache-dir tensorflow==2.12.0 protobuf==3.20.3 wrapt==1.14.0 \
21+
&& pip uninstall tensorflow keras -y
2222

2323
# DLFW Tensorflow packages
2424
COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorflow /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorflow/

0 commit comments

Comments
 (0)