dongdv95 / yolov5

60 stars 64 forks source link

GPU not detected #6

Open YeetMasterYeet opened 1 year ago

YeetMasterYeet commented 1 year ago

Hello, i have cloned the repo and i have followed all of the instructions but i face a problem once running the code: my GPU is not detected, which means that the detector runs on the CPU which is much more slower. when i run the code, i get "YOLOv5 v6.2-109-g23701ea Python-3.8.13 torch-1.12.1+cpu CPU" how do i make my GPU detected by torch?