CUDA out of memory解决办法_cuda out of memory怎么办

作者：凡人多烦事01 | 2024-02-16 00:40:41

踩

cuda out of memory怎么办

问题图片
当使用Pytorch GPU进行计算时经常遇到GPU存储空间过满，原因大致有两点：

1.Batch_size设置过大，超过显存空间
解决办法： 减小Batch_size

2.之前程序运行结束后未释放显存
解决办法： 按住键盘上的Win+R在弹出的框里输入cmd，进入控制台, 然后输入命令

nvidia-smi
1

查看GPU的使用情况，和占用GPU资源的程序，我们看到 PID 2492 运行着python占用很大空间
在这里插入图片描述
可以使用如下命令杀死进程

taskkill -PID 进程号 -F 
1

在这里插入图片描述
如果还不能解决：
参考链接：具体结束各个小程序步骤(Linux系统)

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