出售本站【域名】【外链】

万仟 - 轻松建站从此开始!

微壹擎

当前位置: 微壹擎 > 微信群控 > 文章页

一行代码消除 PyTorch 的 CUDA 内存溢出报错,这个 GitHub 项目刚发布就揽星 60

时间:2021-12-18 22:51来源: 作者:admin 点击: 509 次
几多多人用 PyTorch“炼丹”时都会被那个 bug 困扰。CUDA error: out of memory.正常状况下,你得找出当下占显存的没用的步调,而后 kill 掉。假如不止,还需手动调解 batch size 到适宜的大小,有点省事。如今,有人写

(责任编辑:)

------分隔线----------------------------
发表评论
请自觉遵守互联网相关的政策法规,严禁发布色情、暴力、反动的言论。
评价:
表情:
用户名: 验证码:点击我更换图片
发布者资料
查看详细资料 发送留言 加为好友 用户等级: 注册时间:2024-12-25 02:12 最后登录:2024-12-25 02:12
栏目列表
推荐内容