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-supported xgboost.

xgboost 安装在 tensorflow GPU 支持上

xgboost install on tensorflow GPU support

tensorflow

xgboost