您好，请问我将batchsize设置为4（batchsize设置为2训练时间需要8天）就报以下错误：
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 23.70 GiB total capacity; 17.55 GiB already allocated; 145.56 MiB free; 17.79 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
请问是模型参数太多了吗？我自己的数据集每张图片大小只有40k。谢谢！
zzqshishui
commented
9 months ago
您好，请问我将batchsize设置为4（batchsize设置为2训练时间需要8天）就报以下错误： torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 256.00 MiB (GPU 0; 23.70 GiB total capacity; 17.55 GiB already allocated; 145.56 MiB free; 17.79 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF 请问是模型参数太多了吗？我自己的数据集每张图片大小只有40k。谢谢！