安装

使用 pip

对于熟悉 Python 的用户，我们建议在独立的虚拟环境中使用 [pip][pip] 进行安装。如果不熟悉，可以使用我们预构建的 docker 镜像。

pip install ludwig

对于大型或长时间运行的工作负载，可以使用 Ray 在云端或私有计算集群上远程运行 Ludwig。

安装额外包

Ludwig 的可选功能已拆分成子包。请安装您所需的功能

• ludwig[llm] 用于 LLM 依赖。
• ludwig[serve] 用于服务依赖。
• ludwig[viz] 用于可视化依赖。
• ludwig[hyperopt] 用于超参数优化依赖。
• ludwig[distributed] 用于使用 Dask 在 Ray 上进行分布式训练。
• ludwig[explain] 用于预测解释。
• ludwig[tree] 用于 LightGBM 和基于树的模型。
• ludwig[test] 用于运行 ludwig 的集成和单元测试。
• ludwig[benchmarking] 用于 Ludwig 模型基准测试。
• ludwig[full] 用于所有依赖项。

从 git 安装

pip install git+https://github.com/ludwig-ai/ludwig.git

从源码安装

git clone https://github.com/ludwig-ai/ludwig.git
cd ludwig
pip install -e .

使用 devcontainers

Ludwig 支持在 VSCode devcontainers 上进行开发。请参阅 Ludwig 的 devcontainer 文件。

使用预构建的 Docker 镜像

请参阅 Ludwig 的 docker 文档。