看到这个提示,表示您的GPU内存不足。由于我们经常在PyTorch中处理大量数据,因此很小的错误可能会迅速导致程序耗尽所有GPU; 好的事,这些情况下的修复通常很简单。这里有几个常见检查事项包括: 一、不要在循环训练中累积历史记录。 默认情况下,涉及需要求导/梯度gradients变量的计算将保存在内存中。计算中避免使用这些变量,例如在跟踪统计数据时,这些变量在循环训练中将超出你内存。相反,您...
©️2020 CSDN皮肤主题: 大白 设计师:CSDN官方博客返回首页
zhuhaowCSDN认证博客专家CSDN认证企业博客