1. 程式人生 > >WIN10 xgboost gpu配置最簡便方法

WIN10 xgboost gpu配置最簡便方法

在網上找了很多種方法,結果這種方法是最簡便的,步驟也很簡單。

第一步

這裡下載xgboost原始碼:

第二步

這裡下載

支援GPU版已編譯好的DLL檔案,並放在第一步下載好的檔案xgboost-master/python-package/xgboost目錄下。

第三步

進入到第一步下載好的檔案的xgboost-master/python-package,開啟cmd,執行python setup.py install 命令

至此安裝好了xgboost gpu版本

檢驗安裝是否成功

進入打動\xgboost-master\tests目錄下,開啟cmd,執行 python benchmark_tree.py --tree_method gpu_hist命令

程式能夠順利跑通則說明gpu版本已經安裝成功。

異常情況

1,XGBoost version not compiled with GPU support

將前面安裝好的xgboost解除安裝乾淨後再按照前面步驟重新安裝一遍即可。