admin管理员组

文章数量:1611546

RuntimeError: CUDA out of memory. Tried to allocate 96.00 MiB (GPU 0; 15.75 GiB total capacity; 14.39 GiB already allocated; 51.50 MiB free; 14.71 GiB reserved in total by PyTorch)

把batch_size 设置小一点的数。

本文标签: allocateMIBRuntimeErrorCUDAMemory