After setting up Jetpack 6.1 in my ORIN NX, I have installed the torch from here: https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/
But I am not finding anywhere any compatible torchvision to install with it. Any help with directing me on how to install the correct torchvision package for Jetpack 6.1 would be highly appreciated.
Hi,
Please build it from the source.
You can find the instructions in the topic below:
Thanks.
1 Like
What is the exact version of torchvision for Jetpack 6.1 ?
Python 3.10, CUDA 12.6, torch==torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
It does not work with torchvision main branch, torchvision release/0.20 branch. The error found is below.
import torchvision
Traceback (most recent call last):
File “”, line 1, in
File “/home/nvidia/vision/torchvision/init.py”, line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File “/home/nvidia/vision/torchvision/_meta_registrations.py”, line 164, in
def meta_nms(dets, scores, iou_threshold):
File “/home/nvidia/.local/lib/python3.10/site-packages/torch/library.py”, line 720, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File “/home/nvidia/.local/lib/python3.10/site-packages/torch/library.py”, line 180, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
File “/home/nvidia/.local/lib/python3.10/site-packages/torch/_library/fake_impl.py”, line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, “Meta”):
RuntimeError: operator torchvision::nms does not exist
I also met this question
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.