Skip to content

Unable to setup with GPU/CUDA #255

Answered by ApelSYN
ad-mos asked this question in Q&A
Jan 18, 2023 · 3 comments · 9 replies
Discussion options

You must be logged in to vote

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.

Replies: 3 comments 9 replies

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
9 replies
@ad-mos
Comment options

@ad-mos
Comment options

@ad-mos
Comment options

@ApelSYN
Comment options

@ad-mos
Comment options

Answer selected by ad-mos
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants