How to run a deep learning model on TPU

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
1

I have a jetson Orin AGX. I want to test my DL model on TPU. I tried using pytorch_xla but bumped into several errors. Does any know how to access TPU on Orin ?

3

Hi,

Do you mean GPU or the Tensor Core?
Thanks.

4

I mean Tensor Cores

5

Hi,

You can run TensorRT with the fp16 or int8 flag to deploy the model with Tensor Core.

Thanks.