Skip to content

Latest commit

 

History

History
11 lines (10 loc) · 815 Bytes

install-xgboost-gpu.md

File metadata and controls

11 lines (10 loc) · 815 Bytes

🔑 win10上安装XGBoost-GPU

  1. 找个位置:git clone --recursive https://github.com/dmlc/xgboost
  2. 下载编译文件,并放到xgboost/python-package/xgboost中;
  3. 在xgboost/python-package/目录下运行python setup.py install
  4. 检测cd xgboost/demo/gpu-acceleration/ || python cover_type.py

🔑 Ubuntu上安装XGBoost-GPU

  1. 找个位置:git clone --recursive https://github.com/dmlc/xgboost
  2. cd xgboost, then mkdir build, then cmake .. -DUSE_CUDA=ON and make -j4
  3. cd xgboost/python-package/后,运行sudo apt-get install python-setuptoolssudo -s python setup.py install
  4. 检测cd xgboost/demo/gpu-acceleration\ || python cover_type.py