GPU support on Xavier NX

[barsigor]

Hello,
i have a microk8s cluster with 2 arm SBC and 1 jetson Xavier NX configured as worker node, i try to launch a gpu enable container on xavier, but:

1. cannot enable gpu module, the module is available only for amd64 arch

2. added

default_runtime_name = "nvidia-container-runtime"

in /var/snap/microk8s/current/args/containerd-template.toml

3. created lock dir:

user@jetson1:~$ sudo mkdir -p /var/snap/microk8s/var/lock
user@jetson1:~$ sudo touch /var/snap/microk8s/var/lock/gpu

i tried to reboot and recreate deployment, but the container (a pytorch container from NGC nvidia catalog) cannot see the GPU.

Is there anything else I can try?