Question

我一直在带有3mm数据点和大约40个功能的训练集上在Google Colab上运行一些XGBoost回归。如今，每个回归的运行时间已从约4s增至240s。现在，使用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"
]`

使用XGBoost和“ gpu_hist”的Google Colab突然变慢了

0 个答案: