Hi,
here is the compilation/installation guide for PyTorch (1.7.1) on Drive PX2. Hope it helps!!
Cheers, Vishal Balaji