MarkEz
January 12, 2024, 10:23am
1
Hello,
I have problems with the detection of CUDA:
When I try the following code, the result is always “Not available”:
import cv2
from ultralytics import YOLO
import numpy as np
import torch
if(torch.cuda.is_available()):
print(“Available”)
else:
print(“Not available”)
I want to use YOLOv8, therefore I installed Python 3.8.12 in a virtual environment. I use torch in the version 1.12.1, if I user other version the prediction step with yolo results in a “Illegal instruction” error. The torchvision version is 0.13.0.
Thank you in advance!
Hi,
We only provide the Python 3.6 package for the JetPack 4 environment.
Please build Pytorch from the source with CUDA support if 3.8 is needed.
Please find the building instructions in the below topic:
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 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). You can also use the containers from jetson-containers .
PyTorch pip wheels
JetPack 6
PyTorch v2.2.0 JetPack 6.0 DP (L4T R3…
> Instructions
>> Build from Source
Thanks.
system
Closed
February 13, 2024, 12:30pm
5
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.