Not implemented for use with GPUs


#1

Thought I’d give Gluon a whirl, downloaded it, started up the Jupyter, and part way through the “tutorial”, it get this:

MXNetError: [10:37:33] src/imperative/imperative.cc:78: Operator _ones is not implemented for GPU.

Came from executing this:

z = nd.ones(shape=(3, 3), ctx=mx.gpu(0))

I just re-installed mxnet, and same results. Seems odd to me, that a tool written expressly for use with GPUs, is not implemented for GPU usage.

Any suggestions??

Moldy01


Not implemented for GPU
#2

Did you install the GPU version of MXNet? ‘pip install mxnet’ will give you the CPU version. To get the GPU version, use pip install mxnet-cu90 --pre.


#3

Both times. That was my first thought as well. I thought “I must’ve installed the wrong version”, so I reinstalled. Same results though


#4

Which version of MXNet are you using? Can you copy paste the output of pip show mxnet-cu90?