不同jetson设备对应不同的CMAKE_CUDA_ARCHITECTURES的设置,如下:
# TX1, Nano ------ 53
# TX2 ------ 62
# AGX Xavier, NX Xavier ------ 72
# AGX Orin, NX Orin ------ 87
其他设备:
# K series ------- 30
# M series -------- 50
参考:Issue using Onnxruntime with CUDAExecutionProvider on Orin - #4 by AastaLLL - Jetson AGX Orin - NVIDIA Developer Forums