CUDA out of memory.(已解决)

CUDA out of memory.(已解决)

有时候我们会遇到明明显存够用却显示CUDA out of memory,这时我们就要看看是什么进程占用了我们的GPU。
按住键盘上的Windows小旗子+R在弹出的框里输入cmd,进入控制台。

nvidia-smi

这个命令可以查看GPU的使用情况,和占用GPU资源的程序。我们看到python再运行完以后没有释放资源导致GPU的内存满了。
CUDA out of memory.(已解决)_第1张图片
可以使用如下命令杀死进程,也可以用任务管理器杀死进程。

taskkill -PID 进程号 -F 

在这里插入图片描述
然后再看我们的GPU使用情况,发现内存已经空出来了。这样我们就可以回去执行我们的程序了。
CUDA out of memory.(已解决)_第2张图片

linux系统下

查看占用显存的程序

sudo fuser /dev/nvidia*

CUDA out of memory.(已解决)_第3张图片
使用下面的命令杀死程序

kill -9 pid

你可能感兴趣的:(pytorch)