Resource exhausted: OOM when allocating tensor of shape [4096,8192] and type float

Check what to report here.

You definitely have some problem accessing your GPU.