在CPU和GPU模式下使用LightGBM

Question

我目前在CPU模式下使用LightGBM，并想在GPU模式下试用。CPU和GPU模式的标志不同。有哪些选项可以让我在CPU和GPU模式之间切换？使用conda-env是唯一的选择吗？

Answer 1

查看此处的文档https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html?highlight=gpu#run-你在gpu和GitHub页面上的第一个学习任务：https://github.com/Microsoft/LightGBM/blob/master/docs/Installation-Guide.rst