CUDA out of memory解决办法

CUDA out of memory解决办法

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

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

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

nvidia-smi

查看GPU的使用情况,和占用GPU资源的程序,我们看到 PID 2492 运行着python占用很大空间
CUDA out of memory解决办法_第1张图片
可以使用如下命令杀死进程

taskkill -PID 进程号 -F 

CUDA out of memory解决办法_第2张图片
如果还不能解决:
参考链接:具体结束各个小程序步骤(Linux系统)

你可能感兴趣的:(Pytorch,pytorch,大数据)