在踩过了大部分奇奇怪怪的坑后,终于成功安装了。
以下安装过程是在root用户下安装的,其他用户运行时,可以加上sudo。
安装前提是,Linux上已经安装好GPU需要的驱动等其他安装。
1. 安装GPU支持
使用如下命令安装软件依赖,参考文件中有用统一命令同时安装这些文件,但是我会报错,所以选择一步步安装。
apt-get install ocl-icd-libopencl1
apt-get install ocl-icd-opencl-dev
apt-get install libboost-dev
apt-get install libboost-system-dev
apt-get install libboost-filesystem-dev
2. 安装或更新Python库
同样的原因,选择了一步步安装
pip install setuptools -U
pip install wheel -U
pip install numpy -U
pip install scipy -U
pip install scikit-learn -U
3. 安装LightGBM GPU版本
其中的路径按照自己系统的路径设置就行,官网上也有需要自己cmake的安装步骤,但是按照那样操作最后总会报错,没法写入'compute'目录。自己指定这个opencl的路径安装,既简单也正确,推荐!若安装报错,请查看官方文档,是不是还缺少一些必要的包,如cmake 等。
pip install lightgbm --install-option=--gpu --install-option="--opencl-include-dir=/usr/local/cuda/i