Could you try installing CUDA following these instructions
Also please make sure you use a l4t version of pytorch jetson-containers/packages/l4t/l4t-pytorch at master · dusty-nv/jetson-containers · GitHub
Regards,
Allan Navarro
Embedded SW Engineer at RidgeRun
Contact us: support@ridgerun.com
Developers wiki: https://developer.ridgerun.com/
Website: www.ridgerun.com