I was training a YOLOv8 object detection model on my jetson orin nano, and after the first epoch I got the following error message: AttributeError: partially initialized module ‘torchvision’ has no attribute ‘extension’ (most likely due to a circular import) I realized my torchvision version was…

Hi Please try below for the PyTorch: [image] Install Pytorch with cuda on Jetson Orin nano Devloper Kit Jetson Orin Nano Hi, Please try the below command: $ wget https://developer.download.nvidia.com/compute/redist/jp/v60/pytorch/torch-2.4.0a0+07cecf4168.nv24.05.1471058…

[image] anay.gokhale123: Please help me with this issue Torch vision need to be built from source. Check below link for more info. [image] PyTorch for Jetson Announcements Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavie…

@nagesh_accord But the “Build from Source” section only mentions torch, not torchvision. I tried the torchvision method that is listed under the “Installation” section and that gave me the error that I mentioned in my original post.

Hi, The command for building TorchVision is at “Installation”. Thanks.

@AastaLLL As I mentioned in both my previous messages, Those commands did not work for me. I have put the error message in my initial post. Please help me with that issue

Hi, Which JetPack do you use? For JetPack 6.0, please install the packages in the topics directly to ensure compatibility. PyTorch v2.3.0 JetPack 6.0 (L4T R36.2 / R36.3) + CUDA 12.2 ** torch 2.3 - torch-2.3.0-cp310-cp310-linux_aarch64.whl 1.5k ** torchaudio 2.3 - torchaudio-2.3.0+952ea74-cp3…

Hello, What would the correct command be for the installation of the torchvision and torchaudio using the above packages? The torchvision installation instructions gave me an error as mentioned previously.

Torchvison compatibility

Robotics & Edge Computing Jetson Systems Jetson Orin Nano

AastaLLL August 5, 2024, 5:52am 9

Please try below for the PyTorch:

And below for the TorchAudio and TorchVision:

Thanks.

Topic		Replies	Views
Installing torch and torchvision on Orin Nano Jetson Orin Nano pytorch	2	1575	July 1, 2025
Torchvision version Jetpack 6.0 Jetson Orin Nano pytorch	2	788	August 27, 2024
Pytorch runtime issue on Jetson Orin Nano Jetson Orin Nano pytorch	3	909	June 4, 2024
Pytorch compatibility issues (torch 2.0.0+nv23.5 && torchvision 0.15.1) Jetson Orin NX pytorch	10	18843	June 13, 2023
Torch and torchvision versions are not compatible Jetson AGX Orin pytorch	9	6813	September 11, 2023
Install Pytorch with cuda on Jetson Orin nano Devloper Kit Jetson Orin Nano cuda , pytorch	13	4639	July 30, 2024
Pytorch for jetson nano orin does not work Jetson Orin Nano pytorch	5	1566	August 22, 2023
Jetpack 6.0 Pytorch & Torchvision Compatiblity Error Jetson Orin Nano pytorch	2	1313	January 29, 2024
Jetson Orin Nano (JetPack 5.x): torchvision C++ ops (_C) fail to load with PyTorch 2.0.0+nv23.05 TensorRT jetson	2	58	December 30, 2025
Installing the right torchvision for the torch in jetson orin yolo Jetson Orin Nano yolo	8	868	April 16, 2024

Torchvison compatibility

Related topics