AWS SageMaker MXNet USE_CUDA = 1

Question

Я использую экземпляр AWS ml.p2.xlarge sagemaker и ядро ​​conda_amazonei_mxnet_p36 после установки MXnet CUDA

!pip install mxnet-cu101

при попытке запустить следующий код

mx_tfidf = mx.nd.sparse.array(tfidf_matrix, ctx=mx.gpu())

ЯПолучение следующей ошибки

MXNetError: [19:54:53] src/storage/storage.cc:119: 
Compile with USE_CUDA=1 to enable GPU usage

Пожалуйста, помогите мне решить проблему

nvidia-smi

Answer 1

рассмотрите возможность использования других ядер Jupyter conda_mxnet_p27 или conda_mxnet_p36. Используемое вами ядро ​​- conda_amazonei_mxnet_p36 - в первую очередь предназначено для локального тестирования аппаратного ускорителя Amazon Elastic Inference (отображается как mx.eia() аппаратный контекст в mxnet) и, по-видимому, не поставляется с mx.gpu() контекст включен.