【发布时间】:2022-02-20 21:42:55
【问题描述】:
我在 Jupyter Notebook 中使用 xgboost 包开发了我的模型。 当我们试图将其转移到生产环境时,客户建议使用在他们的环境中可用的 py-xgboost。
但在 Jupyter Notebook 中,我无法导入或安装 py-xgboost 包,因为它不存在。
谁能帮我解决这个问题?
【问题讨论】:
我在 Jupyter Notebook 中使用 xgboost 包开发了我的模型。 当我们试图将其转移到生产环境时,客户建议使用在他们的环境中可用的 py-xgboost。
但在 Jupyter Notebook 中,我无法导入或安装 py-xgboost 包,因为它不存在。
谁能帮我解决这个问题?
【问题讨论】:
我确定你已经尝试过了,但是如果安装了它,在 Jupyter 中,你将 py-xgboost 导入为
import xgboost
如果没有安装,你是在conda还是pip环境?
如果未安装并且您正在使用 pip,您是否尝试过 pip 或 pip3 安装?
pip install xgboost
或
pip3 install xgboost
如果没有安装并且你正在使用 conda,
conda install -c conda-forge xgboost
Sorta tangential,一种非常简单的方法(如果您的 GPU 是 Pascal 或更高版本),您可以在此交互式图形的指导下安装 RAPIDS via conda。它不仅会安装 xgboost 和 py-xgboost,还会让您的整个 pydata 堆栈利用 GPU 加速。
【讨论】:
我也有同样的问题。看来 conda-forge 仓库中的 xgboost 和 py-xgboost 是同一个包,都可以使用
import xgboost as xgb
但是,这两个软件包有纯 CPU 版本,其他版本可能(?)包括 GPU 支持。 xgboost 的 pip (PyPi) 版本包含 GPU 支持,通常比 conda-forge 上的上述两个包更新。
我在 conda 环境中只使用了 pip 版本,但后来我安装了 tpot,它具有 py-xgboost(CPU,例如 1.5.1-cpu_py39ha538f94_2)作为依赖项。所以,现在我在同一个环境中安装了 pip xgboost 和 py-xgboost。当我导入 xgboost 并检查版本时,它与 pip 版本匹配。
获得明确答案的一种方法是在 xgboost github 页面上发布问题。
【讨论】: