xgboost安装在tensorflow GPU支持上

时间:2018-03-09 10:58:43

标签: tensorflow xgboost

我已经安装了tensorflow GPU支持。

尝试通过

在tensorflow上安装xgboost
'conda install -c anaconda py-xgboost' 

我想知道xgboost是不是支持GPU。

我没有安装https://xgboost.readthedocs.io/en/latest/build.html#building-with-gpu-support

只有tensorflow GPU支持。

我是否需要安装xgboost Gpu支持?如果我想使用支持GPU的xgboost

1 个答案:

答案 0 :(得分:1)

您可以检查您的xgboost是否为gpu编译,只是尝试使用tree_method='gpu_hist'或其他gpu方法(here)运行某个模型。 如果它会引发xgboost没有为gpu编译的错误,那么请按照找到的instructions重新安装它。 可能你不需要安装CUDA(如果你已经成功安装了tensorflow-gpu并且它可以运行,那么必须已经安装了CUDA),但你肯定应该构建支持gpu的xgboost。