@Amrit_Das it should work just fine if you use: https://github.com/pytorch/pytorch#from-source 228
I tried it on NVIDIA AGX Xavier which has ARM 64 bit processor, using these commands: https://gist.github.com/soumith/45f61db3379db67414814ad9d5f2a427 69