From 91d53146c89de8b50aa5467a43b9db5028ec5d91 Mon Sep 17 00:00:00 2001 From: Anand Inguva Date: Thu, 8 Jun 2023 14:20:06 -0400 Subject: [PATCH 1/2] Update dockerfile --- .../containers/tensorrt_runinference/tensor_rt.dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile b/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile index 73594cf94d0a..1ac58bab11a3 100644 --- a/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile +++ b/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile @@ -22,8 +22,8 @@ ENV PATH="/usr/src/tensorrt/bin:${PATH}" WORKDIR /workspace -RUN pip install --no-cache-dir apache-beam[gcp]==2.42.0 -COPY --from=apache/beam_python3.8_sdk:2.42.0 /opt/apache/beam /opt/apache/beam +RUN pip install --no-cache-dir apache-beam[gcp]==2.48.0 +COPY --from=apache/beam_python3.8_sdk:2.48.0 /opt/apache/beam /opt/apache/beam RUN pip install --upgrade pip \ && pip install torch>=1.7.1 \ From 7358030c4e5c6e8ca322e696a98eb697325dd654 Mon Sep 17 00:00:00 2001 From: Anand Inguva Date: Thu, 8 Jun 2023 16:22:45 -0400 Subject: [PATCH 2/2] Install venv --- .../containers/tensorrt_runinference/tensor_rt.dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile b/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile index 1ac58bab11a3..04638de50778 100644 --- a/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile +++ b/sdks/python/test-suites/containers/tensorrt_runinference/tensor_rt.dockerfile @@ -33,3 +33,4 @@ RUN pip install --upgrade pip \ && pip install cuda-python ENTRYPOINT [ "/opt/apache/beam/boot" ] +RUN apt-get update && apt-get install -y python3.8-venv