最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

GPU显存不足怎么办?

网站源码admin39浏览0评论

在训练模型中由于GPU储存空间不足而中断

1. 首先查看GPU内存信息

用了13349/24564 Mib,只剩一半不到。

2. 可以查看能清除的文件

我是之前训练过模型,现在想重新训练,所以可以将之前保存的模型删除。

在代码中找到模型保存的位置,rm进行删除

3. 再次查看GPU使用情况

发布评论

评论列表(0)

  1. 暂无评论