我正在使用4GB的NVidia GeForce GTX 1050 ti训练 ResNet ，图像大小为 512x320，使用 6 作为批量大小，这不是很好。
它显示了这个错误：
(1) Resource exhausted: OOM when allocating tensor with shape[6,128,256,160] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
我试图理解张量参数，第一个（6）显然是纪元数，第三个和最后一个（256,160）是图像的维度除以 2。
我很困惑，GPU是否可能没有使用整个内存？