No module named 'torch._custom_ops'

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Orin
4

Hi,

Which TorchVision version do you install?
You can also find this information in the topic you shared:

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

Installation

  • torchvision
    • PyTorch v1.14 - torchvision v0.14.1

Thanks.