Digits only uses cpu when I try to train a model

23 views
Skip to first unread message

Joey Giancola

unread,
Sep 13, 2019, 11:47:00 PM9/13/19
to DIGITS Users
When I am at the home screen it says "1/1 GPU available"
When I create my model it displays my gpu however cpu usage is at 99% while gpu usage never goes above 10%

I am running on debian10 through nvidia-docker

the last few lines of my digits.log in my /jobs/ directory:
2019-09-14 03:35:24 [INFO ] Loaded 1 jobs.
2019-09-14 03:35:40 [20190914-033539-c782] [INFO ] Train Caffe Model task started.
2019-09-14 03:35:40 [20190914-033539-c782] [INFO ] Task subprocess args: "/usr/local/bin/caffe train --solver=/jobs/20190914-033539-c782/solver.prototxt --gpu=0"
2019-09-14 03:35:50 [20190914-033539-c782] [INFO ] Job aborted.
2019-09-14 03:35:52 [20190914-033539-c782] [INFO ] Job deleted.

I suspect I need to change that gpu=0 to gpu=1 but I do not know how.

Thanks for any help 
Reply all
Reply to author
Forward
0 new messages