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/*