MXNetError: GPU not enabled


#1

I am getting the “MXNetError:: GPU not enabled” error on the “collect_params().initialize(ctx=ctx)” line with the following code:

... other stuff...
ctx = mx.gpu(0)
.... other stuff ....
network.collect_params().initialize(ctx=ctx)

Any idea how to fix it?
In case it matters, I am running on Windows after installing MXNet into conda distribution of python using the following commands.

pip install mxnet
pip install mxnet-cu80==1.0.0

I have installed CuDnn on my machine by downloading from NVidia site and the GPU on my machine supports Cuda.


#2

Can we verify outside of MXNet that everything is configured correctly for the GPU. Can you run the command nvidia-smi from the command line. By default it gets installed at C:\Program Files\NVIDIA Corporation\NVSMI