GPU running

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
6

Hi,

YES. The pyTorch provided in this comment is built with GPU support.
So when you train the model on Jetson, it will use GPU acceleration by default.

To check the GPU usage, you can run this command to monitor the system

$ sudo tegrastats

RAM 2283/7763MB (lfb 313x4MB) … GR3D_FREQ 0%@905 APE 75 MTS fg 0% …

Thanks.