热门标签
热门文章
- 1Android在一个APP中通过包名或类名启动另一个APP_android 通过packagename,classname startactivity
- 2取消linux密码复杂性,Ubuntu修改密码及密码复杂度策略设置方法
- 32020 年 中英文拼写纠错开源框架梳理_youngcorrector
- 4OpenGL:VisualStudio2022配置OpenGL环境_opengl配置vs2022
- 5FFMpeg 获取音频音量、提高音量
- 6Sora引领AI生成模型新篇章，视频人工AI创造新视野
- 7OpenCV 图像处理 ------ 白平衡算法_opencv白平衡算法
- 8Android--Service之AIDL传递复杂对象_aidl 复杂对象
- 9阿里数据总监分享《阿里数据中台建设实践案例》，PPT 语音讲解！
- 10linux fork() 进程树的两种实现方法_fork进程树
当前位置: article > 正文
pytorch模型提示超出内存RuntimeError: CUDA out of memory._htts://wisdomai.xyz/tool/pytorch/archives/2301
作者：花生_TL007 | 2024-03-11 20:47:11
赞
踩
htts://wisdomai.xyz/tool/pytorch/archives/2301
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够
简单总结一下解决方法：
- 将batch_size改小。
- 取torch变量标量值时使用item()属性。
- 可以在测试阶段添加如下代码：
with torch.no_grad(): # 停止自动反向计算梯度
- 1
参考：
https://wisdomai.xyz/tool/pytorch/archives/2301
https://ptorch.com/news/160.html
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/220577?site
推荐阅读
相关标签
