Can I try tao toolkit in AGX Orin?

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Hi,

I just bought a AGX Orin, can I try to train pose using the Tao Toolkit on Orin?
I am trying to look for the tutorial but no luck so far.
From this link, it said, “TAO Toolkit is supported on discrete GPUs, such as A100, A40, A30, A2, A16, A100x, A30x, V100, T4, Titan-RTX and Quadro-RTX.” Does it not include Orin? Thx
TAO Toolkit Quick Start Guide — TAO Toolkit 3.22.05 documentation (nvidia.com)

For the pose training, this is the link I just found. Please let me know if there is any video tutorial also, Thx
Training and Optimizing a 2D Pose Estimation Model with NVIDIA TAO Toolkit, Part 1 | NVIDIA Technical Blog

Thx

2

Nope.
TAO is designed on training on the platform with x86-based CPU and discrete GPUs.

After training, user can deploy to model to run inference on discrete GPUs or any Jetson devices(for example, Orin, Xaiver, etc)

2 Likes