yolo dose not use my GPU [closed]

asked 2018-11-05 21:55:29 -0600

pouya.1991
81 ●3 ●9

Hi there, I've installed CUDA 10 on ubuntu 18.04 and compiled both Opencv and darknet with cuda support successfully, but when I wanna to train a yolo network, darknet uses CPU to train the net, what I should do? Is there any other configuration for darknet to use GPU?

edit retag flag offensive reopen merge delete

Closed for the following reason question is off-topic or not relevant by berak
close date 2018-11-06 00:07:05.578775

Comments

As I know darknet has different exe file in GPU and CPU version. Did you use right file?

gino0717 ( 2018-11-05 22:55:05 -0600 )edit

we cannot help you to configure darknet from here.

(and it's only using opencv (optionally) for image acquisition / visualization)

berak ( 2018-11-06 02:33:35 -0600 )edit

add a comment

Comments

so, we have an answer, at least ;)

berak ( 2018-11-06 04:06:45 -0600 )edit

yeah and don't forget to enable CUDNN=1 and OPENCV=1 in the makefile too.

holger ( 2020-04-08 06:53:40 -0600 )edit

add a comment

yolo dose not use my GPU [closed]

Closed for the following reason question is off-topic or not relevant by berak
close date 2018-11-06 00:07:05.578775

Comments

1 answer

Comments

Links

Question Tools

Stats

Related questions

yolo dose not use my GPU [closed] edit

Closed for the following reason question is off-topic or not relevant by berak close date 2018-11-06 00:07:05.578775

Comments

1 answer

Comments

Links

Question Tools

Stats

Related questions

yolo dose not use my GPU [closed]

Closed for the following reason question is off-topic or not relevant by berak
close date 2018-11-06 00:07:05.578775