I’m trying to get mxnet to work on Google Colab. I’m using the
bert-embedding library which uses mxnet, just in case that’s of help. I have ran
!pip instet-cu102all mxn explicitly too, even though bert-embeddings installs it, on Colab and had it imported using
import mxnet as [name].
But, to no avail, I keep getting this error message:
MXNetError: [21:52:28] src/ndarray/ndarray.cc:1279: GPU is not enabled
Here is where I ‘call’ the gpu:
from bert_embedding import BertEmbedding import mxnet as mx ctx = mx.gpu() bert_emb = BertEmbedding(dataset_name='wiki_multilingual_cased', ctx=ctx) # This is the line that throws the error
Like mentioned above, the line that throws the error is when I create a BertEmbedding object, but it’s ultimately a
MXNetError. I’ve tried installing CUDA 10.0 or downgrading the MXNet version to
...-cu100 but I get the same error over and over again.
Platform: Google Colab w/ GPU MXNet Version: Bert installs v1.4.0, I then explicitly tried mxnet-cu100 to -cu102 CUDA version: Tried v10.1 and v10.0
All help is appreciated