Loading [MathJax]/extensions/TeX/extpfeil.js
会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
blackx
博客园
首页
新随笔
联系
管理
订阅
2018年11月5日
pytorch 在测试阶段,进行前向推断时,显存累加溢出问题
摘要: 1、问题描述: pytorch中,在测试阶段进行前向推断运行时,随着for循环次数的增加,显存不断累加变大,最终导致显存溢出。 2、解决方法: 使用如下代码处理输入数据: 假设X为模型的输入 X = X.cuda() input_blobs = Variable(X, volatile=True)
阅读全文
posted @ 2018-11-05 18:13 blackx
阅读(4475)
评论(0)
推荐(0)
编辑
公告
点击右上角即可分享