From 06d6e21de813f3cd8665a82d004cd9d0138a2423 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Fri, 28 Feb 2025 13:48:08 -0700
Subject: [PATCH] Fix cuda targetarch (#16869)

---
 docker/tensorrt/Dockerfile.base | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/tensorrt/Dockerfile.base b/docker/tensorrt/Dockerfile.base
index 5ae018773..4305f1d74 100644
--- a/docker/tensorrt/Dockerfile.base
+++ b/docker/tensorrt/Dockerfile.base
@@ -8,6 +8,7 @@ ARG TRT_BASE=nvcr.io/nvidia/tensorrt:23.12-py3
 # Build TensorRT-specific library
 FROM ${TRT_BASE} AS trt-deps
 
+ARG TARGETARCH
 ARG COMPUTE_LEVEL
 
 RUN apt-get update \
@@ -25,7 +26,6 @@ RUN if [ "$TARGETARCH" = "amd64" ]; then \
 
 # Frigate w/ TensorRT Support as separate image
 FROM deps AS tensorrt-base
-ARG TARGETARCH
 
 #Disable S6 Global timeout
 ENV S6_CMD_WAIT_FOR_SERVICES_MAXTIME=0