最近试用我软的lightgbm,并和xgboost比较了一下。
lightgbm本身带有gpu支持,为了兼容性,使用的是libOpenCL,如果系统没有的话,可以安装ocl-icd-libopencl1 和 ocl-icd-opencl-dev。
nvidia的显卡安装cuda后,本身是带有libOpenCL的,如果再安装ocl-icd-libopencl1 和 ocl-icd-opencl-dev 的话,就要删除nvidia自带的了。
为了兼容性,可以手动编译lightgbm,指定libOpenCL的位置。
1 | mkdir build ; cd build |
编译完成后,如果需要python支持,可以直接安装。1
python setup.py install --precompile
enjoy it!