Pytorch installation Issue

Robotics & Edge Computing Jetson & Embedded Systems Jetson Xavier NX
2

Hi @arsalanmajid116, please install one of the pip wheels from this topic:

These wheels are built with CUDA/cuDNN enabled.

Or you can use the l4t-pytorch or l4t-ml containers from NGC if that’s easier, both of which come with PyTorch and torchvision pre-installed.