admin管理员组

文章数量:1611563

在使用Pytorch中,出现CUDA out of memory. Tried to allocate 392.00 MiB (GPU 0; 10.76 GiB total capacity; 652.77 MiB alread

这是内存不够用的结果,将num_workers和batch size设置小点

本文标签: MIBGPUCUDAMemoryallocate