Mathematica中文论坛-非官方

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 3619|回复: 0

豆粑粑 pytorch for out of memory

[复制链接]

511

主题

578

帖子

2882

积分

金牌会员

Rank: 6Rank: 6

积分
2882
发表于 2023-7-29 10:43:50 | 显示全部楼层 |阅读模式
用pytorch训练,验证模型,
在test阶段,通过迭代实现预测。总是out of memory.
提示显存不够,
2023年7月。A100,32G显示,也经常性的就不够。


    with torch.no_grad():
        [loc_pred,out_prev]= model(input)


可以解决。
回复

使用道具 举报

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

本版积分规则

QQ|Archiver|手机版|小黑屋|Mathematica中文论坛-非官方 ( 辽ICP备16001491号-1

GMT+8, 2023-12-9 02:40 , Processed in 0.126479 second(s), 26 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

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