-
Beta Was this translation helpful? Give feedback.
Replies: 3 comments 9 replies
-
Tried even running docker GPU image ( |
Beta Was this translation helpful? Give feedback.
-
We connect YOLO with "default" settings and then change them. In previous versions of yolo you could specify 'cuda', now you have to write the GPU number, for example '0'
We will improve it in future versions. |
Beta Was this translation helpful? Give feedback.
-
This is taking too long to do, I'll go over the settings tomorrow. |
Beta Was this translation helpful? Give feedback.
We connect YOLO with "default" settings and then change them. In previous versions of yolo you could specify 'cuda', now you have to write the GPU number, for example '0'
You can change the settings for your environment in this file:
pipes/number_plate_localizators/yolo_v5_detector.py
python model = torch.hub.load(repo_path, 'custom', device="0", path=weights, source="local")
We will improve it in future versions.