Hi,
bert_embedding = BertEmbedding(mx.gpu(0))
causes the following error.
Exception has occurred: MXNetError
[15:02:22] C:\Jenkins\workspace\mxnet\mxnet\src\ndarray\ndarray.cc:1295: GPU is not enabled
Please help
best regards,
Cartik
Environment
pip install mxnet-cu90
Run in python code
import mxnet as mx
from bert_embedding import BertEmbedding
bert_embedding = BertEmbedding(mx.gpu(0))
causes error
Exception has occurred: MXNetError
[15:02:22] C:\Jenkins\workspace\mxnet\mxnet\src\ndarray\ndarray.cc:1295: GPU is not enabled