码迷,mamicode.com
首页 > 其他好文 > 详细

ResourceExhaustedError: OOResourceExhaustedError: OOM when allocating tensor with shapeM when allocating tensor with shape

时间:2021-03-16 13:56:03      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:environ   解决   when   cpu   device   sha   string   dev   错误   

原因:是因为gpu显存太小而导致的错误。

解决方法:

1.将batch_size改小

2.不用gpu训练,改用cpu训练

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

ResourceExhaustedError: OOResourceExhaustedError: OOM when allocating tensor with shapeM when allocating tensor with shape

标签:environ   解决   when   cpu   device   sha   string   dev   错误   

原文地址:https://www.cnblogs.com/lpfeng/p/14535506.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!