I am using the AWS ml.p2.xlarge sagemaker instance and conda_amazonei_mxnet_p36 kernel after install MXnet CUDA
!pip install mxnet-cu101
when I try to run the following code
mx_tfidf = mx.nd.sparse.array(tfidf_matrix, ctx=mx.gpu())
I am getting the following error
MXNetError: [19:54:53] src/storage/storage.cc:119:
Compile with USE_CUDA=1 to enable GPU usage
Please help me to resolve the issue
nvidia-smi