按照Github mxnet中的命令pip install --upgrade mxnet==1.4.1 gluonts
安装的mxnet仅是CPU模式。这一模式在训练大量数据(如我所遇到的4.5万行左右)的数据,且用内置的DeepAR模式时,就比较慢了。
按照Install Tutorial中的提示命令pip install mxnet-cu92
安装的话,默认应该是安装了mxnet的最新版本1.5.1。安装好后,运行example,会遇到类似MXNetError while running tutorial examples #246 描述的错误。根据描述仍然需要安装1.4.1的版本。
MXNetError: vector::_M_range_insert```
搜索了一下,找到在阿里云上有whl文件Links for mxnet-cu101,下载安装即可。由于我的CUDA版本是10.1,因此cu后面是101。安装命令:
pip install your-download-file-path