To put it simply, use --gpus
on x86 and --runtime=nvidia
on jetson/ARM SBSA system.
On Jetson, CUDA and some device nodes are shared with the host. You can view this file for more information.
/etc/nvidia-container-runtime/host-files-for-container.d/l4t.csv
For newer Jetpack, please check /etc/nvidia-container-runtime/host-files-for-container.d/drivers.csv
In fact, more details need to refer to nvidia-container-toolkit
, I don’t know much about it