Update Dockerfile

bounverif · May 20, 2024 · c794dfa · c794dfa
1 parent 96c4b82
commit c794dfa
Showing 1 changed file with 10 additions and 7 deletions.
diff --git 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/*