运行代码时出现cuda out of memory吧啦吧啦的错误,搜索发现是显卡内存不足,需要释放。
首先查看显卡状态:nvidia-smi
然后看一下PID那一列,小的那一部分应该是显示屏的显卡,大的那一部分应该是以前运行神经网络所占用的内存,只释放大的那一部分kill -9 PID
PID:换成大的那一部分的数字
然后回车 就成功将内存释放了!
在运行代码就不会出错了~
显卡cuda out of memory
最新推荐文章于 2025-05-28 08:14:07 发布
运行代码时出现cuda out of memory吧啦吧啦的错误,搜索发现是显卡内存不足,需要释放。
首先查看显卡状态:nvidia-smi
然后看一下PID那一列,小的那一部分应该是显示屏的显卡,大的那一部分应该是以前运行神经网络所占用的内存,只释放大的那一部分kill -9 PID
PID:换成大的那一部分的数字
然后回车 就成功将内存释放了!
在运行代码就不会出错了~