Model training with gpu on host machine in xavier agx

I want to do model training with yolov5 on the host machine. I can’t use the gpu despite training with the cpu. In some places, it is said that I need to upgrade because my cuda version is low. (current cuda==10.2.300) please help me urgently on this issue.

I am using 18.4. I just reset jetson xavier. 11.1 OS is not compatible with my system?

will you answer me?