How to use GPU of NVIDIA Xavier NX Developer Kit

I am unable to use GPU for executing YOLO model. I have following configuration:
Machine:aarch64
System:Linux
Distribution: Ubuntu 18.04 Bionic Beaver
Release: 4.9.337-tegra
Python:3.8.16
CUDA Arch BIN: 7.2
L4T:32.7.4
Jetpack:4.6.4

Libraries:
CUDA:10.2.300
cuDNN:8.2.1.32
TensorRT:Missing
VPI:1.2.3
Vulkan:1.2.70
OpenCN:4.5.1 with CUDA:YES

Kindly help me to use GPU with above dependancy.

Hi, YOLO has a lot of implementations. Which one are you trying to run?

YOLOv5

Which framework did you use for your YOLO implementation? Was it TensorFlow, PyTorch, or something else?

PyTorch

This post shows how to install PyTorch for Jetson with GPU support:

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.