Jetson nano gpu accelerators

Hi,
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: https://developer.download.nvidia.com/compute/redist/jp/v46/tensorflow/

And there are JetPack 4 wheels for PyTorch here:

There are also various containers with GPU support here: