python-libnvinfer is finally delivered properly on ARM64.

NVIDIA-Merlin · Sep 13, 2023 · e2c9eff · e2c9eff
1 parent b1497a3
commit e2c9eff
Showing 1 changed file with 1 addition and 5 deletions.
diff --git 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/*