e-lab/ENet-training

cuda runtime error (10) : invalid device ordinal

Closed this issue · 1 comments

Hi, is there any solution to this error?

==> defining some tools

==> flattening model parameters THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-4230/cutorch/init.c line=719 error=10 : invalid device ordinal

==> defining some tools

==> flattening model parameters /home/CV/torch/install/bin/luajit: /home/CV/torch/install/share/lua/5.1/trepl/init.lua:384: ...V/torch/install/share/lua/5.1/cunn/DataParallelTable.lua:634: cuda runtime error (10) : invalid device ordinal at /tmp/luarocks_cutorch-scm-1-4230/cutorch/init.c:719 stack traceback:

`[C]: in function 'error'
/home/CV/torch/install/share/lua/5.1/trepl/init.lua:384: in function 'require'
run.lua:56: in main chunk

[C]: in function 'dofile'
.../CV/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk

[C]: at 0x00406670`

Many thanks!

The issue is solved by adding --nGPU to 1.