xgboost安装在tensorflow GPU支持上

Question

我已经安装了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

Answer 1

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