Error in pytorch & torchvision on Xavier NX JP 5.0.1 DP

[kun.he.love.u]

Hi,
I installed pytorch and torchvision on Xavier NX JP 5.0.1 DP following the instructions in the link below:

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Python on Jetson Nano, Jetson TX1/TX2, Jetson Xavier NX/AGX, and Jetson AGX Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). PyTorch pip wheels

PyTorch v1.12.0

JetPack 5.0 (L4T R34.1.0) / JetPack 5.0.1 (L4T …

but I get below warning:

import torchvision
/home/hekun/Downloads/torchvision/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
/home/hekun/Downloads/torchvision/torchvision/init.py:28: UserWarning: You are importing torchvision within its own root folder (/home/hekun/Downloads/torchvision). This is not expected to work and may give errors. Please exit the torchvision project source and relaunch your python interpreter.
warnings.warn(message.format(os.getcwd()))
import torch
print(torch.version)
1.11.0
print('CUDA available: ’ + str(torch.cuda.is_available()))
CUDA available: True
print('cuDNN version: ’ + str(torch.backends.cudnn.version()))
cuDNN version: 8302
print(torchvision.version)
0.12.0

And I get the following error when running yolov5 training:

File “/home/hekun/.local/lib/python3.8/site-packages/torchvision/extension.py”, line 33, in _assert_has_ops
raise RuntimeError(
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 GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision 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.

Could you help me? Thanks a lot!

[AastaLLL]

Hi,

Could you check if the same error occurs when using l4t-pytorch:r34.1.0-pth1.12-py3 container below?

NVIDIA L4T PyTorch | NVIDIA NGC

Thanks.

[brbl]

I am running into the same issue: the container works fine but the download doesn’t. It seems like there is a missing dependency that is picked up in the container but not in the download. Nosing around, it seems as if this is torch/torchvision issue (it appears elsewhere) and not an nvidia problem as such.

[brbl]

Ok, for what it is worth, what I wound up doing is to download and install pytorch per the instructions and then built the latest version of torchvision from source. That seemed to cure the problem as of today :-)