Skip to content

Commit b387366

Browse files
author
youngday
committed
cuda test ok
1 parent 9358afe commit b387366

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

README.md

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,19 @@ cargo run -r --example yolo -- --task pose --ver 8 --scale n --model ../models/v
8686
cargo run -r --example yolo -- --task pose --ver 11 --scale n --model ../models/v11-m-pose.onnx --source assets/bus.jpg
8787
```
8888

89-
rtmo
90-
89+
### rtmo
90+
```sh
9191
cargo run -r --example rtmo -- --model ../models/rtmo/m.onnx --source assets/bus.jpg
92+
```
93+
### cuda
94+
95+
after install cuda and cudnn,
96+
97+
add:
98+
99+
--device cuda:0
100+
101+
92102

93103
### other
94104

0 commit comments

Comments
 (0)