Update cudaqx-dev image to include tensorrt

Signed-off-by: Ben Howe <[email protected]>

Author: bmhowe23

docker/build_env/cudaqx.dev.Dockerfile

@@ -22,6 +22,17 @@ RUN apt-get update && CUDA_DASH=$(echo $cuda_version | tr '.' '-') \
   && python3 -m pip install "cmake<4" --user \
   && apt-get autoremove -y --purge && apt-get clean && rm -rf /var/lib/apt/lists/*
 
+# Install tensorrt-dev, unless this is CUDA 12 w/ arm64
+RUN CUDA_MAJOR_VERSION=$(echo $cuda_version | cut -d . -f1); \
+  ARCH="${ARCH:-$(dpkg --print-architecture)}"; \
+  if [ "$CUDA_MAJOR_VERSION" != "12" ] || [ "$ARCH" = "amd64" ]; then \
+    apt-get update \
+    && apt-cache search tensorrt | awk '{print "Package: "$1"\nPin: version *+cuda'${cuda_version}'\nPin-Priority: 1001\n"}' | tee /etc/apt/preferences.d/tensorrt-cuda${cuda_version}.pref > /dev/null \
+    && apt update \
+    && apt-get install -y tensorrt-dev \
+    && apt-get autoremove -y --purge && apt-get clean && rm -rf /var/lib/apt/lists/*; \
+  fi
+
 COPY .cudaq_version /cudaq_version
 
 ENV CUDAQ_INSTALL_PREFIX=/usr/local/cudaq