Tensorflow运行时ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape

爱做梦的病人

于 2019-02-22 23:21:38 发布

阅读量3.7k

点赞数

文章标签： python tensorflow ResourceExhaustedErro

本文链接：https://blog.youkuaiyun.com/qq_40431912/article/details/87887100

版权

使用Tensorflow库进行cifar-10模型测试时，出现以下错误：

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[10000,32,32,32] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
	 [[Node: Conv2D = Conv2D[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Conv2D-0-TransposeNHWCToNCHW-LayoutOptimizer, Variable/read)]]
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.

	 [[Node: Mean/_21 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1,