From 3249d460fc24e5a1fa649d9e40c19cb36cac0ec5 Mon Sep 17 00:00:00 2001 From: qqiao Date: Wed, 26 Jun 2024 06:24:22 -0700 Subject: [PATCH] Update for new torch --- docker/dockerfile.merlin.ctr | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/dockerfile.merlin.ctr b/docker/dockerfile.merlin.ctr index 9af6ffdab..d797702fc 100644 --- a/docker/dockerfile.merlin.ctr +++ b/docker/dockerfile.merlin.ctr @@ -261,8 +261,7 @@ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow backe # Torch Metrics and Lightning (without torch) RUN pip install --no-cache-dir tensorflow==2.15.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 \ + && pip install --no-cache-dir --no-deps torchmetrics \ && rm -rf /usr/local/lib/python${PYTHON_VERSION}/dist-packages/caffe2 # DLFW Tensorflow packages @@ -285,6 +284,7 @@ RUN ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/ # DLFW Python packages COPY --chown=1000:1000 --from=torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch COPY --chown=1000:1000 --from=torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch.dist-info/ +COPY --chown=1000:1000 --from=torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torchgen /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torchgen COPY --chown=1000:1000 --from=torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorrt /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorrt COPY --chown=1000:1000 --from=torch /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorrt-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/tensorrt.dist-info/