Hello, i am trying to run a yolo model on GPU on jetson agx 64gb but it does not work. right now its running on cpu. when i tried to run it on gpu, i have this error “AssertionError: Torch not compiled with CUDA enabled”. I checked and realized that CUDA was not installed in my jetson yet, so could you send me to some links that have CUDA installation instructions? i also try to install opencv with CUDA, so could you send me to some links to do that?
I am moving this topic to the Jetson category for greater visibility.
Hi @trannhuquynh1406, you can find PyTorch wheels for Jetson that were compiled with CUDA enabled here:
To build OpenCV with CUDA, you can use this script, or find links to pre-built binaries used by my containers: