V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
V2EX 提问指南
faller
V2EX  ›  问与答

深度学习 1660s AMP 加速问题

  •  
  •   faller · 230 天前 · 327 次点击
    这是一个创建于 230 天前的主题,其中的信息可能已经有所发展或是发生改变。

    使用 apex amp 后内存占用下降,但时间是原来的 5 倍, 前 5 耗时如下,切换 O1 O2 会导致其发生变化, 应该?换卡?!

    ncalls tottime percall cumtime percall filename:lineno(function)
    82 342.244 4.174 342.244 4.174 {method 'item' of 'torch._C._TensorBase' objects}
    91411 65.138 0.001 65.138 0.001 {method 'decode' of 'ImagingDecoder' objects}
    1280 8.153 0.006 8.153 0.006 {imread}
    80 1.464 0.018 1.464 0.018 {method 'run_backward' of 'torch._C._EngineBase' objects}
    1280 0.654 0.001 0.654 0.001 {built-in method PIL._imaging.new}
    目前尚无回复
    关于   ·   帮助文档   ·   API   ·   FAQ   ·   我们的愿景   ·   广告投放   ·   感谢   ·   实用小工具   ·   2354 人在线   最高记录 5497   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 25ms · UTC 15:28 · PVG 23:28 · LAX 07:28 · JFK 10:28
    ♥ Do have faith in what you're doing.