From e2c9eff7c7a44bcaab120e64c8276ca6209f0967 Mon Sep 17 00:00:00 2001
From: Matthias Langer <bashimao@users.noreply.github.com>
Date: Wed, 13 Sep 2023 05:48:31 -0700
Subject: [PATCH] `python-libnvinfer` is finally delivered properly on ARM64.

---
 docker/dockerfile.merlin | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
index fd73a21ef..77e08edbe 100644
--- a/docker/dockerfile.merlin
+++ b/docker/dockerfile.merlin
@@ -190,6 +190,7 @@ RUN ARCH=$([ "${TARGETARCH}" = "arm64" ] && echo "sbsa" || echo "x86_64") && \
         python3 \
         python3-pip \
         python3-dev \
+        python3-libnvinfer \
         rapidjson-dev \
         tree \
         wget \
@@ -211,11 +212,6 @@ RUN ARCH=$([ "${TARGETARCH}" = "arm64" ] && echo "sbsa" || echo "x86_64") && \
         openssh-server \
         # [ HugeCTR ]
         libaio-dev && \
-        # NOTE: libnvinfer is installed anyway, just Python bindings are missing on ARM.
-    if [[ "$TARGETARCH" != "arm64" ]]; then \
-        # TensorRT dependencies
-        apt install -y --no-install-recommends python3-libnvinfer \
-    ; fi && \
     apt autoremove -y && \
     apt clean && \
     rm -rf /var/lib/apt/lists/*