You are not logged in. Your edit will be placed in a queue until it is peer reviewed.
We welcome edits that make the post easier to understand and more valuable for readers. Because community members review edits, please try to make the post substantially better than how you found it, for example, by fixing grammar or adding additional resources and hyperlinks.
Required fields*
-
1I'm having the same issue. I've trained the model with GPU enabled (yolo detect train model=yolov8m.pt data=data/pasxalitses.yaml imgsz=640 workers=8 batch=16 device=0 epochs=3000 name=yolov8_pasx_t100m) and I've set the model to run on the GPU but it keeps using the CPU instead. I'm working on a Jetson Orin AGX developer kit, with CUDA enabled both for OpenCV and PyTorch. is there any other way to enable the GPU?Christina– Christina2023-04-12 11:22:23 +00:00Commented Apr 12, 2023 at 11:22
-
updated my answer with some more docs - maybe those will help219CID– 219CID2023-04-12 13:08:48 +00:00Commented Apr 12, 2023 at 13:08
-
When using your solution {model = YOLO("yolov8n.pt", device='gpu')} I get the same error you had as well. I've tried model = YOLO("dnn_model/yolov8l.pt") and model.to(device), while setting the device to cuda but it doesn't seem to work either. The model.predict didn't work either, as it took more than 1min to run. For the model.predict I set the device to device=0 since everything else throws an exception.Christina– Christina2023-04-12 13:27:27 +00:00Commented Apr 12, 2023 at 13:27
Add a comment
|
How to Edit
- Correct minor typos or mistakes
- Clarify meaning without changing it
- Add related resources or links
- Always respect the author’s intent
- Don’t use edits to reply to the author
How to Format
-
create code fences with backticks ` or tildes ~
```
like so
``` -
add language identifier to highlight code
```python
def function(foo):
print(foo)
``` - put returns between paragraphs
- for linebreak add 2 spaces at end
- _italic_ or **bold**
- indent code by 4 spaces
- backtick escapes
`like _so_` - quote by placing > at start of line
- to make links (use https whenever possible)
<https://example.com>[example](https://example.com)<a href="https://example.com">example</a>
How to Tag
A tag is a keyword or label that categorizes your question with other, similar questions. Choose one or more (up to 5) tags that will help answerers to find and interpret your question.
- complete the sentence: my question is about...
- use tags that describe things or concepts that are essential, not incidental to your question
- favor using existing popular tags
- read the descriptions that appear below the tag
If your question is primarily about a topic for which you can't find a tag:
- combine multiple words into single-words with hyphens (e.g. python-3.x), up to a maximum of 35 characters
- creating new tags is a privilege; if you can't yet create a tag you need, then post this question without it, then ask the community to create it for you