cuda not available in jetson Nano

naseemap47 / streamlit-yolo

Display predicted Video, Images and webcam using YOLO models (YOLOv7 & YOLOv8) with Streamlit

MIT License

34 stars 22 forks source link

cuda not available in jetson Nano #21

Open Mirshal opened 1 week ago

Mirshal commented 1 week ago

I'm trying to run through the GPU, I have all the cuda equipment installed properly but it is constrained in cuda no available, is there a certain version of cuda to be used in this code?

naseemap47 commented 1 week ago

Hi @Mirshal, Can you check this ?

python3 -c "import torch;print(torch.cuda.is_available());print(torch.cuda.get_device_name())"

Mirshal commented 1 week ago

thanks, this problem done to solving, but I have a problem when applying to a 16GB nvidia jetson nano orin nx system. The cuda is running, but CUDNN has problems displaying object counting statistics.

naseemap47 commented 1 week ago

Hi @Mirshal, JetPack will come with CUDA, TensorRT and cuDNN. I think there is no chance for a problem. Can you share more details about the problem, if possible.