问题描述
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 (将#修改为@)