python常见错误：显卡内存不够

最新推荐文章于 2024-04-28 10:23:25 发布

费马罗尔拉格朗日柯西……

最新推荐文章于 2024-04-28 10:23:25 发布

阅读量1.6k

点赞数 1

文章标签： 1024程序员节

本文链接：https://blog.csdn.net/hm_uin/article/details/127487293

版权

tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[32,1024,1,1024] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
	 [[{{node gradients/AddN_7/tmp_var}} = TemporaryVariable[dtype=DT_FLOAT, shape=[32,1024,1,1024], var_name="gradients/AddN_7/tmp_var", _device="/job:localhost/replica:0/task:0/device:GPU:0"](^gradients/conv5/bn/batchnorm/mul_1_grad/Sum_1, ^gradients/conv5/bn/batchnorm/mul_1_grad/Reshape)]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

解决办法：减小batch的大小

或修改图片大小

或者确定显卡内存在哪一层不够，若是全连接层则降低修改全连接层维度