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