Installing pytorch on jetson tx2

[smilesajid14]

@dusty_nv
Ubuntu 18.04
jetpack 4.6.5
L4T Version 32.7.5
Python 3.6.9

I download the wheel from here Index of /compute/redist/jp/v461/pytorch

when i am trying to install torch 1.11 but it is installing torch 1.8.0. I have created a virtual environment also.

Before installing i tried following commands

pip3 uninstall torch
pip3 cache purge

[EduardoSalazar96]

Hi @smilesajid14

Could you please check this post PyTorch for Jetson?

Regards!
Eduardo Salazar
Embedded SW Engineer at RidgeRun

Contact us: support@ridgerun.com
Developers wiki: https://developer.ridgerun.com/
Website: www.ridgerun.com

[smilesajid14]

Hi @EduardoSalazar96

Thanks for your reply.

But I don’t find any pytorch wheel compatible with jetpack 4.6.5 and L4T version 32.7.5 in this link PyTorch for Jetson

That is why I have used following link to download wheel .
https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/

[EduardoSalazar96]

Hi @smilesajid14

I understand. However, what happens if you try using PyTorch v1.10.0? If I were in your situation, I would give it a try.

Maybe someone on the NVIDIA side could clarify which PyTorch version you can use.

Regards!

[smilesajid14]

Thanks for your reply but when i am trying to install Pytorch v 1.10.0 it still installs the Pytorch v 1.8.0 instead .

[AastaLLL]

Hi

What is the checksum of the Pytorch 1.11.0 downloaded in your environment?

$ md5sum torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl 
f90802fa30323829dc4902e5a2b957fa  torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl

Thanks.