maxnet src/ndarray/ndarray.cc:1279: GPU is not enabled

运行下吗代码出现标题中的错误。

from bert_embedding import BertEmbedding
import mxnet as mx
ctx = mx.gpu()
bert_emb = BertEmbedding(dataset_name='wiki_multilingual_cased', ctx=ctx) 

我的环境：cuda10.0 mxnet-cu100 1.9.0

解决方法：

从mxnet 官网发下 cuda10.0对应的 mxnet 版本最高是1.8.0，但是我直接安装上的是1.9.0。

pip uninstall mxnet-cu100
pip install mxnet-cu100==1.8.0

再次运行代码，如果遇到ImportError: libopenblas.so.0: cannot open shared object file，安装下 libopenblas 即可。

apt-get update
apt-get install libopenblas-dev

没问题啦！

import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()
array([[ 3.,  3.,  3.],
       [ 3.,  3.,  3.]], dtype=float32)