Neural Network training in AGX Xavier

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
1

Hello there,

Does the Jetson Xavier AXG supports backpropagration? can we train a neural network in the Xavier?

Regards,

2

Hi,

This depends on the framework you use.

In our TensorRT library, we don’t support backpropagation since TensoRT is designed for inference only.
However, it’s still possible to use TensorFlow for training on Jetson.
https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html

But please noticed that we still recommend to get a desktop GPU for the training task.
Jetson is designed as an edge device so the memory and bandwidth is not be suitable for training.

Thanks.