翻译 OOM when allocating tensor with shape[60000,32,28,28] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:Conv2D]
时间: 2023-05-29 21:01:52 浏览: 321
解决TensorFlow GPU版出现OOM错误的问题
在分配形状为[60000,32,28,28]和类型为float的张量时,在分配器GPU_0_bfc中,由于/ job:localhost / replica:0 / task:0 / device:GPU:0上的Conv2D操作,发生OOM。
阅读全文