1

在 GPU 和 CPU 上运行 LightGBM 的 Higgs 训练对我来说花费了相同的时间 - 26 秒。日志确认 GPU 运行正在使用 GPU(将数据传输到 GPU 等)

https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html

通过教程为 GPU 安装 LightGBM。安装良好并且能够运行 GPU 训练来确认。

下面的 GPU 和 CPU 规格进行比较。

Processor (Skylake X; Latest Generation)
8-Core 3.60 GHz Intel Core i7-7820X

VS

NVIDIA RTX 2080 8 GB

注意:按照教程接受所有配置默认值,因为这是基准。可以和他们一起玩,但这可能会破坏这一点。

有人在RTX上尝试过类似的吗?

性能基准https://lightgbm.readthedocs.io/en/latest/GPU-Performance.html正在使用 GTX,但表明任何最近的 NVIDIA 卡确实应该可以工作。基准页面警告 Kepler 卡。但RTX 2080是图灵,似乎支持硬件原子操作。https://en.wikipedia.org/wiki/CUDA我有限的知识表明它应该都很好。

4

0 回答 0