I couldn’t find which version of torchvision is compatible for Pytorch version (Torch Version: 2.5.0a0+872d972e41.nv24.08).
I clonned vision repo for torchvision installation and I installed 0.20.1 version for torchvision but I got this error:
(env) dalab@ubuntu:~/vision$ cd …
(env) dalab@ubuntu:~$ python3
Python 3.10.12 (main, Nov 6 2024, 20:22:13) [GCC 11.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.
import torch
import torchvision
Traceback (most recent call last):
File “”, line 1, in
File “/home/dalab/env/lib/python3.10/site-packages/torchvision/init.py”, line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File “/home/dalab/env/lib/python3.10/site-packages/torchvision/_meta_registrations.py”, line 164, in
def meta_nms(dets, scores, iou_threshold):
File “/home/dalab/env/lib/python3.10/site-packages/torch/library.py”, line 720, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File “/home/dalab/env/lib/python3.10/site-packages/torch/library.py”, line 180, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
File “/home/dalab/env/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
print(“Torch Version:”, torch.version)
Torch Version: 2.5.0a0+872d972e41.nv24.08
print(“Torchvision Version:”, torchvision.version)
Traceback (most recent call last):
File “”, line 1, in
NameError: name ‘torchvision’ is not defined
print(“CUDA Available:”, torch.cuda.is_available())
CUDA Available: True
if torch.cuda.is_available():
… print(“CUDA Device:”, torch.cuda.get_device_name(0))
…
CUDA Device: Orin