MXNet error related to CUDA
suvam-bag opened this issue · comments
Hi @itijyou,
On executing either the validation set/training set test I get the following error -
Check failed: e == cudaSuccess CUDA: invalid device ordinal
I have bene unsuccessful at resolving this error. The MNIST example from mxnet runs okay with GPU.
It seems like this error is coming from setting the wrong device_id for the GPU.
Any help in resolving this would be appreciated.
I am trying to run it on a machine with a single Nvidia GTX 1080.