Torchvision for JetPack 6?

Following Installing PyTorch for Jetson Platform - NVIDIA Docs, I found a whl for JetPack 6.0dp here (https://developer.download.nvidia.cn/compute/redist/jp/v60dp/pytorch/), which seems like it work for the production release of JetPack 6.0. But all I can find for torchvision are the instructions to build from source here (PyTorch for Jetson).

So now I’m building torchvision from source. Is there really no pre-built whl, either from pytorch.org or from nvidia? Difficulties with versions and manual compile steps seem to be pretty common in the forums.

Thanks.

Hi @nathanielfairfield, you can find pre-built torchvision wheels for JetPack 6 and CUDA 12.2 here:

http://jetson.webredirect.org/jp6/cu122

http://jetson.webredirect.org/jp6/cu122/+simple/torchvision/

Thank you @dusty_nv ! (BTW, building from source did work for me, too).

A small suggestion would be to add references to this from Installing PyTorch for Jetson Platform - NVIDIA Docs for the next person :)

1 Like

Thanks @nathanielfairfield - noted! 👍 for the time being, have updated this thread with the latest: PyTorch for Jetson