GPU drives on jetson nano

Autonomous Machines Jetson & Embedded Systems Jetson Nano
1

This picture is trining googlenet network condition. Does the GPU drive? If GPU doesn’t drive,how can I handle it? I use this web(jetson-inference/pytorch-collect.md at master · dusty-nv/jetson-inference · GitHub) to train.

InkedIMG_8501_LI
InkedIMG_8501_LI1477×1108 408 KB

3

Hi,

Suppose you are confused for the GPU driver.

The GPU driver for Jetson device is integrated into the OS.
So the driver is installed after flashing the device from sdkmanager.

The log from your image is fine.
It indicates that pytorch is using GPU index==0 for training.

Thanks.