Jetson nano gpu accelerators

I’ve developed python algorithms that use traditional ML models(sklearn and tensorflow) and have transferred my algorithms to a jetson nano. what libraries/implementations can i use to activate the gpu and accelerate the algorithms? I have looked into Rapids but understand it won’t work on a jetson nano. Looking for something high-level.

Hi @jacobostapenko10, you can find TensorFlow wheels for JetPack 4.6 with CUDA support here:

And there are JetPack 4 wheels for PyTorch here:

There are also various containers with GPU support here: