Hello, I am quite new to GPU processing and the Jetson Products.
I am currently training a machine learning model on a Jetson AGX Xavier, but it is taking way too long been waiting for about 24 hours right now. I just realized I might not be making use of the GPU, meaning that I am probably running on CPU that is why it’s taking so long.
My question then is how do I execute scripts on Xavier on GPU rather than CPU?
In general, we don’t recommend Jetson platform for training.
Jetson is designed as an edge platform so the memory and bandwidth might not be the optimal for training task.