Xgboost实现GPU加速
生活随笔
收集整理的這篇文章主要介紹了
Xgboost实现GPU加速
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
環境配置
首先最好你有帕斯卡構架的GPU,如果沒有的話,最好還是不要搞GPU加速了。
首先安裝CUDA
第二步:
下載xgboost源碼:
https://github.com/dmlc/xgboost
第三步:
從http://ssl.picnet.com.au/xgboost/中下載支持GPU版已編譯好的DLL文件
第四步:
將xgboost.dll復制到xgboost-master/python-package/xgboost目錄下,
打開控制臺,進入到python-package目錄下,執行python setup.py install命令,
至此xgboost GPU版安裝成功。
需要注意的是,需要先把源碼文件放到類似于這樣的c:\users\keked\appdata\local\programs\python\python36\lib\site-packages路徑里面。
總結
以上是生活随笔為你收集整理的Xgboost实现GPU加速的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 如何给一个空白磁盘安装操作系统
- 下一篇: 关于魔法方法的一点总结