Skip to content

Commit

Permalink
Remove install of protobuf since we move it from dlfw
Browse files Browse the repository at this point in the history
  • Loading branch information
EmmaQiaoCh committed Jun 11, 2024
1 parent 5b7a131 commit 68dfccb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docker/dockerfile.merlin.ctr
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake<
tensorflow-metadata \
"scikit-learn<1.2" \
tritonclient[all]
RUN pip install --no-cache-dir protobuf==3.20.3 onnx onnxruntime==1.15.1 pycuda
RUN pip install --no-cache-dir onnx onnxruntime==1.15.1 pycuda
RUN pip install --no-cache-dir onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com

# Triton Server
Expand Down Expand Up @@ -258,7 +258,7 @@ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow backe
# Need to install transformers after tensorflow has been pulled in, so it builds artifacts correctly.
# upgrade to 4.36.0 due to GHSA-3863-2447-669p
# Torch Metrics and Lightning (without torch)
RUN pip install --no-cache-dir tensorflow==2.14.0 protobuf==3.20.3 wrapt==1.14.0 transformers==4.36.0 \
RUN pip install --no-cache-dir tensorflow==2.14.0 wrapt==1.14.0 transformers==4.36.0 \
&& pip uninstall tensorflow keras -y \
&& pip install --no-cache-dir --no-deps torch torchmetrics \
&& rm -rf /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch \
Expand Down

0 comments on commit 68dfccb

Please sign in to comment.