jfrank94 commented on issue #20363: URL: https://github.com/apache/incubator-mxnet/issues/20363#issuecomment-863621233
@TristonC The docker image version is 19.3.12, but it's running through a Kubernetes cluster on Azure. In order to reproduce this error, you should download these packages first on a Jupyter notebook: !sudo apt install libquadmath0 !sudo apt install libnccl2=2.8.4-1+cuda11.2 --allow-downgrades -y !pip install mxnet-cu112 !pip install --upgrade gluonts !pip install scipy !pip install xlrd==1.2.0 Then import the mxnet library as well as these libraries under GluonTS: from gluonts.dataset.common import ListDataset from gluonts.model.deepar import DeepAREstimator from gluonts.mx.trainer import Trainer from gluonts.dataset.util import to_pandas from gluonts.evaluation import Evaluator from gluonts.evaluation.backtest import make_evaluation_predictions from gluonts.support.util import get_download_path from gluonts.mx.trainer import Trainer from gluonts.model.predictor import Predictor Then, you can import a dataset and convert it into a ListDataSet like so: training_data = ListDataset( [{"start": train.index[0], "target": train.value}], freq = "1H" ) Finally, initialize the DeepAREstimator and train it like so: estimator = DeepAREstimator(freq="1D", prediction_length=28, trainer=Trainer(epochs=60, ctx=ctx, num_batches_per_epoch=1), num_parallel_samples=1) train1_output = estimator.train(training_data=training_data, validation_data=validation_data) I hope this helps! -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org For additional commands, e-mail: issues-h...@mxnet.apache.org