Check failed: exec_ctx.dev_id < device_count_ (1 vs. 1)

jessemelpolio / Faster_RCNN_for_DOTA

Code used for training Faster R-CNN on DOTA

https://arxiv.org/abs/1711.10398

Apache License 2.0

336 stars 102 forks source link

Check failed: exec_ctx.dev_id < device_count_ (1 vs. 1) #9

Closed Shamsa279 closed 6 years ago

Shamsa279 commented 6 years ago

while running test file I got this error I know its GPU error but I dont know how to resolve it can you help me out? mxnet.base.MXNetError: [13:48:30] src/engine/threaded_engine.cc:320: Check failed: exec_ctx.dev_id < devicecount (1 vs. 1) Invalid GPU Id: 1, Valid device id should be less than device_count: 1

jessemelpolio commented 6 years ago

If you have Nvidia GPUs, the starting number should be 0. You may set the gpu_id to 0 if you only have one gpu.