CUDA out of memory.(已解决)

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

nvidia-smi

这个命令可以查看GPU的使用情况,和占用GPU资源的程序。我们看到python再运行完以后没有释放资源导致GPU的内存满了。
在这里插入图片描述
可以使用如下命令杀死进程,也可以用任务管理器杀死进程。

taskkill -PID 进程号 -F 

在这里插入图片描述
然后再看我们的GPU使用情况,发现内存已经空出来了。这样我们就可以回去执行我们的程序了。
在这里插入图片描述

linux系统下

查看占用显存的程序

sudo fuser /dev/nvidia*

在这里插入图片描述
使用下面的命令杀死程序

kill -9 pid
Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐