CMAKE_CUDA_ARCHITECTURES针对Jetson Xavier或者Orin的设置

不同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

CUDA GPU | NVIDIA Developer

你可能感兴趣的:(深度学习,人工智能)