
解决TensorFlow GPU版本的OOM问题
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文章介绍了如何解决使用TensorFlow GPU版本时遇到的内存溢出(OOM)问题,提供了多种有效的方法和建议。
在使用mask_rcnn预测自己的数据集时,会出现以下错误:ResourceExhaustedError: OOM when allocating tensor with shape [1,512,1120,1120] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[{{node rpn_model/rpn_conv_shared/convolution}} = Conv2D[T=DT_FLOAT, data_format=NCHW, dilation
全部评论 (0)
还没有任何评论哟~


