这个屌丝很懒，什么也没留下！

解决cuda out of memory的几种方法_cudaoutofmemory怎么办

作者：小蓝xlanll | 2024-03-18 13:54:10

cudaoutofmemory怎么办

1测试的时候不需要梯度反向传播时，使用

with torch.no_grad()

2在计算累计损失时，使用

total_loss+=loss.item()

将tensor转为非tensor

3.在保证batchsize尽可能大的条件下，调小batchsize

torch.cuda.empty_cache()

nividia-smi

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/263514