From c794dfa509f52da977f0a17a1c48d21b30e9698f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Do=C4=9Fan=20Ulus?= <dogan.ulus@bogazici.edu.tr>
Date: Mon, 20 May 2024 13:30:16 +0300
Subject: [PATCH] Update Dockerfile

---
 containers/autoware-build/Dockerfile | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/containers/autoware-build/Dockerfile b/containers/autoware-build/Dockerfile
index 7bdc1d7..970b87e 100644
--- a/containers/autoware-build/Dockerfile
+++ b/containers/autoware-build/Dockerfile
@@ -42,13 +42,16 @@ RUN wget ${CUDA_KEYRING_FILEPATH} && dpkg -i ${CUDA_KEYRING_PACKAGE} && rm ${CUD
 
 RUN apt-get update && export DEBIAN_FRONTEND=noninteractive \
     && apt-get -y install \
-      cuda-minimal-build-12-3 \
-      libcublas-dev-12-3 \
-      libcurand-dev-12-3 \
-      libnvinfer-dev \
-      libnvinfer-plugin-dev \
-      libnvparsers-dev \
-      libnvonnxparsers-dev \
+        cuda-nvcc-12-4 \
+        libcublas-dev-12-4 \
+        libcurand-dev-12-4 \
+        libcusolver-dev-12-4 \
+        libnvinfer-dev=8.* \
+        libnvinfer-headers-dev=8.* \
+        libnvinfer-plugin-dev=8.* \
+        libnvinfer-headers-plugin-dev=8.* \
+        libnvparsers-dev=8.* \
+        libnvonnxparsers-dev=8.* \
     && find . -name 'libcu*.a' -delete \
     && find . -name 'libnv*.a' -delete \ 
     && apt-get autoremove -y && apt-get clean -y && rm -rf /var/lib/apt/lists/*