Caffe: Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal

Created on 28 Jul 2016  路  2Comments  路  Source: weiliu89/caffe

when i try to train SSD, the shell command is "python examples/ssd/ssd_pascal.py "
The system is Ubuntu 14.04, with one GPU
Can anyone give me some advice, thank you.
@weiliu89

picture fuxianh

Most helpful comment

You should change this line to gpus = '0'

picture weiliu89 on 28 Jul 2016
👍3

All 2 comments

You should change this line to gpus = '0'

weiliu89 picture weiliu89 on 28 Jul 2016
👍3

Wow, so great, thank you. Next time I would check the train file first.

fuxianh picture fuxianh on 28 Jul 2016
Was this page helpful?
0 / 5 - 0 ratings

Related issues

tumitx picture tumitx  路  5Comments
songjmcn picture songjmcn  路  5Comments
siddharthm83 picture siddharthm83  路  4Comments
lcj1105 picture lcj1105  路  3Comments
Coderx7 picture Coderx7  路  3Comments