From 2cf577078596f6b1ba79a41ff8877332d274d44d 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 | 8 ++++---- 2 files changed, 5 insertions(+), 5 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 b02026ea26f..b4e557bef48 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.6.3-devel-ubuntu22.04 FROM ${AARCH64_BASE_IMAGE} ENV DEBIAN_FRONTEND=noninteractive \ - CUDA_CROSS_VERSION=11-8 \ - CUDA_CROSS_VERSION_DOT=11.8 + CUDA_CROSS_VERSION=12-6 \ + CUDA_CROSS_VERSION_DOT=12.6 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 && \ @@ -33,7 +33,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/ python3.12 python3.12-dev \ python3.13 python3.13-dev && \ apt-key adv --fetch-key http://repo.download.nvidia.com/jetson/jetson-ota-public.asc && \ - add-apt-repository 'deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/cross-linux-aarch64/ /' && \ + add-apt-repository 'deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/cross-linux-aarch64/ /' && \ apt-get update && \ apt-get install -y cuda-cudart-cross-aarch64-${CUDA_CROSS_VERSION} \ libcufft-cross-aarch64-${CUDA_CROSS_VERSION} \