File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff 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
8081ARG _HUGECTR_BACKEND_REPO="github.com/triton-inference-server/hugectr_backend.git"
8182ARG TRITON_VERSION
8283# Install Triton inference backend.
Original file line number Diff line number Diff 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
2424COPY -- chown = 1000 : 1000 -- from = dlfw / usr/ local/ lib/ python${PYTHON_VERSION}/ dist- packages/ tensorflow / usr/ local/ lib/ python${PYTHON_VERSION}/ dist- packages/ tensorflow/
You can’t perform that action at this time.
0 commit comments