I have successfully been able to get Yolov5 working on my Jetson Xavier NX. However, I realized that the software is not using any of my GPU memory. After some research, I realized I installed the PyTorch version that was for the CPU and not the GPU. The thing is, installing pytorch was incredibly d…

Hi @bryantsp , please see this topic for pre-built PyTorch wheels for Jetson with CUDA support: [image] PyTorch for Jetson Jetson Nano Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of …

Installing PyTorch for CUDA 10.2 on Jetson Xavier NX for YOLOv5

Robotics & Edge Computing Jetson Systems Jetson Xavier NX

dusty_nv July 16, 2021, 6:21pm 2

Hi @bryantsp, please see this topic for pre-built PyTorch wheels for Jetson with CUDA support:

3 Likes

Topic		Replies	Views
Can I execute yolov5 on the GPU of JETSON AGX XAVIER? Jetson AGX Xavier yolo	22	7911	July 1, 2022
Unable to use GPU with pytorch yolov5 on jetson nano Jetson Nano cuda , pytorch	2	2801	October 20, 2022
Pytorch & torchversion compatible issue on L4T35.5.0 Jetson Orin Nano pytorch	20	1415	November 7, 2024
Jetson Xavier NX RuntimeError during inference Pytorch YOLOv8 Jetson Xavier NX pytorch	3	152	January 28, 2025
Pytorch compatibility issues (torch 2.0.0+nv23.5 && torchvision 0.15.1) Jetson Orin NX pytorch	10	18701	June 13, 2023
Jetson Xavier Nx - Cuda toolkit Jetson Xavier NX opencv , cuda , jetson-inference	3	758	June 7, 2023
Torchvision for jetson xavier nx with jetpack 5.1 Jetson AGX Xavier pytorch	4	2752	February 21, 2023
YOLOv5 on Jetson Orin not use GPU Jetson AGX Orin yolo	4	1934	November 9, 2022
Install Pytorch with cuda on Jetson Orin nano Devloper Kit Jetson Orin Nano cuda , pytorch	13	4475	July 30, 2024
I would like to ask if i can train a new model by NX? Jetson Xavier NX ai-training	41	2411	October 18, 2021

Installing PyTorch for CUDA 10.2 on Jetson Xavier NX for YOLOv5

Related topics