安装
使用pip或conda安装NeuralForecast
您可以通过以下命令从 Python 包索引 安装 NeuralForecast
的 发布版本:
pip install neuralforecast
或者
conda install -c conda-forge neuralforecast
Tip
神经预测方法可以利用 GPU 计算。确保已安装 Cuda。
Warning
我们正在不断更新 neuralforecast,因此建议您固定版本以避免问题。 pip install neuralforecast=="1.0.0"
Tip
我们建议您在 Python 虚拟环境或 conda 环境 中安装库。
附加功能
您可以使用以下附加功能来增加可选功能:
- 使用 spark 进行分布式训练:
pip install neuralforecast[spark]
- 从 S3 保存和加载:
pip install neuralforecast[aws]
使用我们的环境(可选)
如果您没有 Conda 环境并需要 Numba、Pandas、NumPy、Jupyter、Tune 和 Nbdev 等工具,您可以通过以下步骤使用我们的环境:
- 克隆 NeuralForecast 仓库:
$ git clone https://github.com/Nixtla/neuralforecast.git && cd neuralforecast
- 使用
environment.yml
文件创建环境:
$ conda env create -f environment.yml
- 激活环境:
$ conda activate neuralforecast
- 安装 NeuralForecast 开发版
$ pip install -e ".[dev]"
Give us a ⭐ on Github