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

在一个 gpu 中同时训练 50 个卷积神经网络求助

  •  
  •   Sonwi · 2021-12-28 21:03:43 +08:00 · 1197 次点击
    这是一个创建于 1061 天前的主题,其中的信息可能已经有所发展或是发生改变。

    计算资源只有一个 12g 显存的 2080ti ,需要同时训练 50 个卷积神经网络( 50 个进程),网络结构很简单,两个卷积池化层加两个全连接层,数据集是 cifar10 ,怎样才能充分利用这个 gpu 又不超显存呢

    2 条回复    2021-12-28 22:47:39 +08:00
    ruxuan1306
        1
    ruxuan1306  
       2021-12-28 22:18:26 +08:00
    少进程大批次,多进程小批次,反正积是常量显存大小。

    但多进程可能切换代价高,大批次可能随机下降慢。
    ipwx
        2
    ipwx  
       2021-12-28 22:47:39 +08:00
    这里建议你多买几张卡
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1040 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 22ms · UTC 19:30 · PVG 03:30 · LAX 11:30 · JFK 14:30
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.