微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

LightGBM 基准测试显示 RTX-2080 GPU 上的 CPU 没有加速

如何解决LightGBM 基准测试显示 RTX-2080 GPU 上的 CPU 没有加速

LightGBM 的 Higgs 训练运行对我来说在 GPU 和 cpu 上花费的时间相同 - 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 卡都应该确实可以工作。基准页面对开普勒卡发出警告。但是RTX 2080是图灵的,似乎支持硬件原子操作。 https://en.wikipedia.org/wiki/CUDA 我有限的知识表明它应该都很好。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。