want to use GPU for YOLO8, but no way to install TORCH properly. Maybe I do something wrong? YOLO is to use with camera and detect/follow objects. Any idea?
Hello,
I use
pip3 install torch-2.6.0rc1-cp310-cp310-linux_aarch64.whl
pip3 install torchaudio-2.5.0-cp310-cp310-linux_aarch64.whl
pip3 install torchvision-0.20.0-cp310-cp310-linux_aarch64.whl
All goes well without errors.
When I check with
import torch
print(torch.cuda.is_available()) # Should return True
print(torch.backends.cudnn.version()) # Should return a version number
I get