I found troubling trying to use my GPU in Jetson Nano. I have tried both, Tensorflow and Pytorch installed from Nvidia Docs. The code is simple ‘Hello AI World’ type, in short no problem with it. However, when I run it, it stuck in training. And it does not move on anywhere and Jetson just does not operate, i.e. closing terminal or rebooting. Nevertheless, when I run on CPU mode, everything works just fine.
Things I tried:
- Latest Jetpack and Tensorflow installed from Nvidia Docs.
- Latest Jetpack and PyTorch installed from Nvidia Docs.
- Latest Jetpack and downgraded PyTorch (1.7.0, 1.8.0)
- Downgraded Jetpack and latest PyTorch…
In short, whatever I do, it is impossible to train on GPU.
I don’t know where problem is. Could this be due to Power supply? (Jetson is connected to USB cable and travel adapted charger with 9 V and 1.67 A or 5 V and 2.0 A output.)
I would be grateful for any clues.