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:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
