While running my code in NVIDIA AGX Xavier™ device ( JetPack v4.6.3), I am getting the following error:
File "/home/ravi/.local/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 102, in _batched_nms_vanilla curr_keep_indices = nms(boxes[curr_indices], scores[curr_indices], iou_threshold) File "/home/ravi/.local/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 34, in nms _assert_has_ops() File "/home/ravi/.local/lib/python3.6/site-packages/torchvision/extension.py", line 63, in _assert_has_ops "Couldn't load custom C++ ops. This can happen if your PyTorch and " RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check https://github.com/pytorch/vision#installation for the compatibility matrix. Please check your PyTorch version with torch.__version__ and your torchvision version with torchvision.__version__ and verify if they are compatible, and if not please reinstall torchvision so that it matches your PyTorch install.
Based on the official documentation, it seems that torch v1.9.0 is compatible with torchvision v0.10.0. But I am still getting the error. Please see more info. below:
$ ipython3 Python 3.6.9 (default, Nov 25 2022, 14:10:45) Type 'copyright', 'credits' or 'license' for more information IPython 7.16.3 -- An enhanced Interactive Python. Type '?' for help. In : import torch In : import torchvision In : torch.cuda.is_available() Out: True In : torch.__version__ Out: '1.9.0' In : torchvision.__version__ Out: '0.10.0'
Please suggest a workaround to get torchvision working.