RuntimeError: operator torchvision::nms does not exist

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
5

Hi,

Thanks, it looks like there are some issues with PyTorch 2.6.0rc1 and TorchVision 0.20.0.
Please use the PyTorch 2.5.0 and TorchVision 0.19.1 instead (compatible to JetPack 6.1):

$ wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/5cf/9ed17e35cb752/torch-2.5.0-cp310-cp310-linux_aarch64.whl#sha256=5cf9ed17e35cb7523812aeda9e7d6353c437048c5a6df1dc6617650333049092 -O torch-2.5.0-cp310-cp310-linux_aarch64.whl
$ pip3 install torch-2.5.0-cp310-cp310-linux_aarch64.whl 
$ wget https://pypi.jetson-ai-lab.dev/jp6/cu126/+f/9d2/6fac77a4e832a/torchvision-0.19.1a0+6194369-cp310-cp310-linux_aarch64.whl#sha256=9d26fac77a4e832a3a2a547543aabc6256b6f94b6531f6a94a762989db4535f8 -O torchvision-0.19.1a0+6194369-cp310-cp310-linux_aarch64.whl
$ pip3 install torchvision-0.19.1a0+6194369-cp310-cp310-linux_aarch64.whl

We can import both PyTorch and TorchVision without issues:

$ python3 
Python 3.10.12 (main, Sep 11 2024, 15:47:36) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.__version__
'2.5.0'
>>> torchvision.__version__
'0.19.1a0+6194369'
>>>

Thanks.

3 Likes