GPU not detected on jetson xavier nx devkit

Hi. I am running this host image:

cat resin-boot/os-release  
ID="balena-os"
NAME="balenaOS"
VERSION="2.82.11+rev8"
VERSION_ID="2.82.11+rev8"
PRETTY_NAME="balenaOS 2.82.11+rev8"
MACHINE="jetson-xavier-nx-devkit"
VARIANT="Development"
VARIANT_ID=dev
META_BALENA_VERSION="2.82.11"

I tried to build my image locally but it fails with the following error:

[Build]   main   gpu id :0 not exist !
[Build]   main   Loading pre-trained weights...
[Build]   main   Weights loading complete!
[Build]   main   ERROR: CUDA initialization failure with error 100. Please check your CUDA installation:  http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
[Build]   main   Segmentation fault (core dumped)

It seems like the gpu is not visible.

Running command balena run -it --rm --gpus all balenalib/jetson-xavier-nx-devkit-ubuntu:bionic-build bash on the host, I get the following error:

balena: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].

I might be missing something but the gpus should be visible, no? How can I trouble shoot this?

Thanks,

Hello @le13francois

Did you manage to resolve this issue?

I believe you might require the io.balena.features.gpu=1 on the respective containers. Can you please give it a try?

Also, you might find this blog post helpful.

Hope this helps

Thanks