canonical / microk8s

MicroK8s is a small, fast, single-package Kubernetes for datacenters and the edge.
https://microk8s.io
Apache License 2.0
8.53k stars 773 forks source link

GPU support on Xavier NX #4461

Open barsigor opened 8 months ago

barsigor commented 8 months ago

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?

QingLea commented 2 months ago

Same issue, no idea how to see gpu in microk8s on Jetson devices.