时间戳

首页 > 解决方案 > 谷歌 Colab 使用带有“gpu_hist”的 XGBoost 突然变慢了

gpu - 谷歌 Colab 使用带有“gpu_hist”的 XGBoost 突然变慢了

问题描述

我一直在 Google Colab 上运行一些 XGBoost 回归，训练集具有 3 毫米数据点和大约 40 个特征。今天，每个回归的运行时间已从大约 4 秒变为 240 秒。使用tree_method="hist"现在比gpu_hist. 请问有其他人经历过吗？

我已经安装了 linux GPU 版本并指定了tree_method="gpu_hist". 我已多次将运行时类型设置和重置为 GPU，但均未成功。GPU内存似乎也不是问题：

`[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 10791916159412462458
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 8329653355733370145
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 8283103013471747914
physical_device_desc: "device: XLA_GPU device"
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 11123789005
locality {
  bus_id: 1
  links {
  }
}
incarnation: 2057146031472419157
physical_device_desc: "device: 0, name: Tesla K80, pci bus id: 0000:00:04.0, compute capability: 3.7"
]`

标签： gpugoogle-colaboratoryxgboost

解决方案

推荐阅读