Is there a compatible version of torchvision for the current version of pytorch available?

Autonomous Machines Jetson & Embedded Systems Jetson Xavier NX
1

Hi,

Nicely, I see that there is a python3 wheel for pytorch available to download here:

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

(This looks to be the latest version, for jetpack 5.0 developer preview)

Is there also a related and compatible version of torchvision available to download as a wheel as well ?

I couldn’t find one.

Cheers,
Kim

2

Actually, I see I posted the link for the jetpack 4.6.1 version but if you have the answer for both versions of jetpack that would be great.

Cheers,
Kim

3

Oh and I see that that wheel for the developer preview jetpack 5.0 is pytorch 1.12 which is listed as latest and untested on the pytorch website, I for one would find the latest stable version to be more practically useful.

Just saying

Kim

4

Hi @KimHendrikse, PyTorch 1.11 for JetPack 5.0 Developer Preview is available here:

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available

PyTorch v1.11.0

5

For torchvision we don’t release the wheels, so you can build it from source like shown in this thread, or you can use the l4t-pytorch container which already has PyTorch / torchvision / torchaudio pre-installed.

6

Okay thanks.

Kim

7

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