GPU running

How to set GPU running?

Hi,

This depends on the package you installed.
If your pyToch is built with GPU support, it should set GPU enabled by default.

To check this, you can monitor the system status with this command:

$ sudo tegrastats

Thanks.

When I use this web(jetson-inference/pytorch-collect.md at master · dusty-nv/jetson-inference · GitHub) to train googlenet, can I use GPU to train network on jetson nano?

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.