Xavier NX torch.cuda.is_available() returns FALSE

Hello guys,

I used yolov5 but my xavier nx is too slow to run models.
I installed pytorch , but when I run this code; Xavier NX torch.cuda.is_available() returns FALSE …!!

Please help me…

you may use the pre-built pytorch thing
as in Torch.cuda.is_available() returns false - #3 by AastaLLL