如何在Python的特定conda环境中使用GPU设置LightGBM？

Question

如何在Python的特定conda环境中安装支持GPU的LightGBM模块？

谢谢

Answer 1

目前只有 https://github.com/microsoft/LightGBM/blob/master/python-package/README.rst 有实际收据，例如对于 Ubuntu 仅适用于

pip install lightgbm --config-settings=cmake.define.USE_GPU=ON

在所需的 conda 环境中。

所有包含

setup.py

的收据现已过时，因为

“...setup.py 从 v4.0.0 开始已从 lightgbm 中删除。”

如开发人员所述 https://github.com/microsoft/LightGBM/issues/6240#issuecomment-1863900158

https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html 已经过时了。

所以目前获得它的唯一方法是

conda install <all other packages that you need in environment>

然后激活它，然后使用上面的pip命令。

对于非linux平台，请参考第一个链接中的收据咨询依赖关系。