PyTorch wheel for JetPack 4.6.1 missing from Nvidia servers

I’m trying to install PyTorch on JetPack 4.6.1 / L4T 32.7.1 which uses Python 3.6.

The documentation at Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation was updated April 28 to prepare for JetPack 5.0 which is still in Developer Preview.

The information for installing on previous versions, even the current release version, is not available anymore.

I was able to guess that the wheel is available from this directory but the download link 404s.

https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/

Where can I get this file?

In lieu of being able to access the .whl my workaround was to docker pull nvcr.io/nvidia/l4t-pytorch:r32.7.1-pth1.10-py3 then copy all torch* files from /usr/local/lib/python3.6/dist-packages/ to my host, and create easy-install.pth for torchvision and torchaudio

This is extremely hacky and not recommended!

I found it on the Jetson Download Center which gives this URL:

https://developer.download.nvidia.com/compute/redist/jp/v461/pytorch/torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl

The broken link is missing the +nv22.01.

1 Like

Hi @rgovostes,

Thanks for pointing this out. I have the team looking to fix the broken link on the documentation site.

Cheers,
Tom K

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.