RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while com

Hi @pardeshi.grishma07, did you compile torchvision like as shown in the Installation section of this topic?

It’s possible that if you pip installed other package (like YOLO ultralytics for example), it may have re-installed the torchvision from pip (which wasn’t built with CUDA enabled). So you may need to re-install torchvision again. You can also use the l4t-pytorch container which already has these packages installed and working.

1 Like