lightgbm使用cuda自带的libOpenCL

原文转载自 「Neo Fung's Blog」 (https://www.neofung.org/2017/09/18/lightgbm使用cuda自带的libOpenCL/)

预计阅读时间 0 分钟(共 0 个字, 0 张图片, 0 个链接)

最近试用我软的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
2
3
mkdir build ; cd build
cmake -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include -DUSE_GPU=1 ..
make -j4

编译完成后,如果需要python支持,可以直接安装。

1
python setup.py install --precompile

enjoy it!

more_vert