From 5afe94926687e2173cffa44011402f3852d6897d Mon Sep 17 00:00:00 2001 From: Janusz Lisiecki Date: Tue, 30 Apr 2024 15:41:52 +0200 Subject: [PATCH] Move to JetPack 6.0 for Xavier - upgrade cross compilation cuda to 12.2 to match JetPack 6.0 - adds support for sm90a (THOR) Signed-off-by: Janusz Lisiecki --- cmake/CUDA_utils.cmake | 2 +- docker/Dockerfile.build.aarch64-linux | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/CUDA_utils.cmake b/cmake/CUDA_utils.cmake index fab2d58ff54..508f9b7f61f 100644 --- a/cmake/CUDA_utils.cmake +++ b/cmake/CUDA_utils.cmake @@ -40,7 +40,7 @@ endfunction() # List of currently used arch values if (${ARCH} MATCHES "aarch64-") # aarch64-linux - set(CUDA_known_archs "53" "62" "72" "75" "87") + set(CUDA_known_archs "53" "62" "72" "75" "87" "90a") elseif (${ARCH} MATCHES "aarch64") # aarch64 SBSA, only >=Volta # from the whole list/; "70" "75" "80" "86" diff --git a/docker/Dockerfile.build.aarch64-linux b/docker/Dockerfile.build.aarch64-linux index e9f93b9ca78..59f50913144 100644 --- a/docker/Dockerfile.build.aarch64-linux +++ b/docker/Dockerfile.build.aarch64-linux @@ -1,9 +1,9 @@ -ARG AARCH64_BASE_IMAGE=nvidia/cuda:11.8.0-devel-ubuntu20.04 +ARG AARCH64_BASE_IMAGE=nvidia/cuda:12.2.2-devel-ubuntu20.04 FROM ${AARCH64_BASE_IMAGE} ENV DEBIAN_FRONTEND=noninteractive \ - CUDA_CROSS_VERSION=11-8 \ - CUDA_CROSS_VERSION_DOT=11.8 + CUDA_CROSS_VERSION=12-2 \ + CUDA_CROSS_VERSION_DOT=12.2 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \ apt-get update && apt-get install software-properties-common -y --no-install-recommends && \