返回顶部
noavatar
  • 发帖数12
  • 粉丝0

此人很懒,什么也没有留下

7*24新情报

💥搞了个骚操作:把Python跑在GPU上,速度起飞!

[复制链接]
Xzongzhi 显示全部楼层 发表于 2 小时前 |阅读模式 打印 上一主题 下一主题
兄弟们,最近折腾了个事儿,真给我整兴奋了。以前写数据处理脚本,几百万条记录,CPU跑得跟蜗牛似的,忍了很久。昨晚突发奇想,试了试NVIDIA的cupy库,直接把numpy代码改两行,丢到GPU上跑。结果?同样的循环,原来要45秒,现在0.8秒就干完了!😱

别误会,不是所有场景都适合,得是那种矩阵运算多的活儿。但如果你跟我一样,天天跟数组、矩阵打交道,真心建议试试。cupy接口跟numpy几乎一模一样,迁移成本低到哭。唯一坑的是,显存不够别硬来,我1060的6G显存,跑个1亿条数据就炸了。

新手别上来就搞这个,先把CPU版本调优再说。老鸟们,还有啥类似的黑科技?求推荐,最近对加速上瘾了。
回复

使用道具 举报

default_avator1
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver·手机版·闲社网·闲社论坛·羊毛社区· 多链控股集团有限公司 · 苏ICP备2025199260号-1

Powered by Discuz! X5.0   © 2024-2025 闲社网·线报更新论坛·羊毛分享社区·http://xianshe.com

p2p_official_large
快速回复 返回顶部 返回列表