在Ubuntu系统下用显卡训练深度学习模型时,总会遇到以下问题:
已放弃(核心已转储)
此时说明GPU的显存已满,一种解决办法是重启命令行或重启电脑,另一种方法是手动清理显存,步骤如下:
首先在命令行输入指令
nvidia-smi
查看显卡使用情况,可以发现最右下角名为python的进程占用显存最多,但实际已将程序关闭,这时就需要手动清理
可以观察到,python进程的PID号为120489,此时执行以下命令:
sudo kill -9 PID # 请执行下面的命令
需注意将PID替换为自己需要清理的进程PID号,如本例中将执行:
sudo kill -9 120489
注意:在清理时要确保所清理的进程已经没有用!