MXNet error related to CUDA

[suvam-bag]

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.