admin管理员组

文章数量:1611942

使用pytorch进行训练是报出以下错误

RuntimeError: CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0; 5.79 GiB total capacity; 4.75 GiB already allocated; 25.38 MiB free; 4.86 GiB reserved in total by PyTorch)

这是因为GPU的内存溢出,可以调小batch_size

本文标签: allocateMIBRuntimeErrorCUDAMemory