python中XGBoost如何選擇GPU設備

Question

我在 Python 的共享服務器 linux 上運行 XGBoost（eXtreme Gradient Boosting），該服務器配備了多個 GPU。 當我在 GPU 上運行 XGBoost 時，XGBoost 會自動選擇 GPU 設備 0。有人知道我如何使 XGBoost 成為 select 另一個 GPU 嗎？ （gpu0 通常太忙了）

Answer 1

查看源代碼，原來可以通過將 gpu 設備號分配給gpu_id來在 class 的構造函數中選擇它。 例如，

xgb.XGBClassifier(tree_method='gpu_hist',predictor='gpu_predictor',gpu_id=1)