Mxnet92 cuda9.2 win10 cannot use gpu


#1

I try to run the validation code on official site, when I import mx, there is a warning

C:****/Anaconda3\lib\site-packages\h5py_init_.py:36: FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated. In future, it will be treated as np.float64 == np.dtype(float).type.
from ._conv import register_converters as _register_converters

when I run this line
a= mx.nd.ones((2,3),mx.gpu())
The kernel appears to have died.

What should I do?


#2

How did you install mxnet? Did you use pip install or build it from source?


#3

I used this command
pip install mxnet-cu92


#4

Could you try using mxnet-cu90 instead? I tried installing mxnet-cu92 on Windows Server 2016 and once I import it, then it fails with OSError: [WinError 126] The specified module could not be found.

mxnet-cu90 worked fine for me though.


#5

Problem solved by install a lower version of mxnet,
pip install mxnet-92==1.2.0