今天在服务器上更新训练模型，五个小时后发现进程被kill了
理解 OOM killer
最近有位 VPS 客户抱怨 MySQL 无缘无故挂掉，还有位客户抱怨 VPS 经常死机，登陆到终端看了一下，都是常见的 Out of memory 问题。这通常是因为某时刻应用程序大量请求内存导致系统内存不足造成的，这通常会触发 Linux 内核里的 Out of Memory (OOM) killer，OOM killer 会杀掉某个进程以腾出内存留给系统用，不致于让系统立刻崩溃。如果检查相关的日志文件（/var/log/messages）就会看到下面类似的 Out of memory: Kill process 信息：
1 ... 2 Out of memory: Kill process 9682 (mysqld) score 9 or sacrifice child 3 Killed process 9682, UID 27, (mysqld) total-vm:47388kB, anon-rss:3744kB, file-rss:80kB 4 httpd invoked oom-killer: gfp_mask=0x201da, order=0, oom_adj=0, oom_score_adj=0 5 httpd cpuset=/ mems_allowed=0 6 Pid: 8911, comm: httpd Not tainted 2.6.32-279.1.1.el6.i686 #1 7 ... 8 21556 total pagecache pages 9 21049 pages in swap cache 10 Swap cache stats: add 12819103, delete 12798054, find 3188096/4634617 11 Free swap = 0kB 12 Total swap = 524280kB 13 131071 pages RAM 14 0 pages HighMem 15 3673 pages reserved 16 67960 pages shared 17 124940 pages non-shared
Linux 内核根据应用程序的要求分配内存，通常来说应用程序分配了内存但是并没有实际全部使用，为了提高性能，这部分没用的内存可以留作它用，这部分内存是属于每个进程的，内核直接回收利用的话比较麻烦，所以内核采用一种过度分配内存（over-commit memory）的办法来间接利用这部分 “空闲” 的内存，提高整体内存的使用效率。一般来说这样做没有问题，但当大多数应用程序都消耗完自己的内存的时候麻烦就来了，因为这些应用程序的内存需求加起来超出了物理内存（包括 swap）的容量，内核（OOM killer）必须杀掉一些进程才能腾出空间保障系统正常运行。用银行的例子来讲可能更容易懂一些，部分人取钱的时候银行不怕，银行有足够的存款应付，当全国人民（或者绝大多数）都取钱而且每个人都想把自己钱取完的时候银行的麻烦就来了，银行实际上是没有这么多钱给大家取的。
内核检测到系统内存不足、挑选并杀掉某个进程的过程可以参考内核源代码 linux/mm/oom_kill.c，当系统内存不足的时候，out_of_memory() 被触发，然后调用 select_bad_process() 选择一个 “bad” 进程杀掉，如何判断和选择一个 “bad” 进程呢，总不能随机选吧？挑选的过程由 oom_badness() 决定，挑选的算法和想法都很简单很朴实：最 bad 的那个进程就是那个最占用内存的进程。
1 /** 2 * oom_badness - heuristic function to determine which candidate task to kill 3 * @p: task struct of which task we should calculate 4 * @totalpages: total present RAM allowed for page allocation 5 * 6 * The heuristic for determining which task to kill is made to be as simple and 7 * predictable as possible. The goal is to return the highest value for the 8 * task consuming the most memory to avoid subsequent oom failures. 9 */ 10 unsigned long oom_badness(struct task_struct *p, struct mem_cgroup *memcg, 11 const nodemask_t *nodemask, unsigned long totalpages) 12 { 13 long points; 14 long adj; 15 16 if (oom_unkillable_task(p, memcg, nodemask)) 17 return 0; 18 19 p = find_lock_task_mm(p); 20 if (!p) 21 return 0; 22 23 adj = (long)p->signal->oom_score_adj; 24 if (adj == OOM_SCORE_ADJ_MIN) { 25 task_unlock(p); 26 return 0; 27 } 28 29 /* 30 * The baseline for the badness score is the proportion of RAM that each 31 * task's rss, pagetable and swap space use. 32 */ 33 points = get_mm_rss(p->mm) + p->mm->nr_ptes + 34 get_mm_counter(p->mm, MM_SWAPENTS); 35 task_unlock(p); 36 37 /* 38 * Root processes get 3% bonus, just like the __vm_enough_memory() 39 * implementation used by LSMs. 40 */ 41 if (has_capability_noaudit(p, CAP_SYS_ADMIN)) 42 adj -= 30; 43 44 /* Normalize to oom_score_adj units */ 45 adj *= totalpages / 1000; 46 points += adj; 47 48 /* 49 * Never return 0 for an eligible task regardless of the root bonus and 50 * oom_score_adj (oom_score_adj can't be OOM_SCORE_ADJ_MIN here). 51 */ 52 return points > 0 ? points : 1; 53 }
理解了这个算法我们就理解了为啥 MySQL 躺着也能中枪了，因为它的体积总是最大（一般来说它在系统上占用内存最多），所以如果 Out of Memeory (OOM) 的话总是不幸第一个被 kill 掉。解决这个问题最简单的办法就是增加内存，或者想办法优化 MySQL 使其占用更少的内存，除了优化 MySQL 外还可以优化系统（优化 Debian 5，优化 CentOS 5.x），让系统尽可能使用少的内存以便应用程序（如 MySQL) 能使用更多的内存，还有一个临时的办法就是调整内核参数，让 MySQL 进程不容易被 OOM killer 发现。
配置 OOM killer
我们可以通过一些内核参数来调整 OOM killer 的行为，避免系统在那里不停的杀进程。比如我们可以在触发 OOM 后立刻触发 kernel panic，kernel panic 10秒后自动重启系统。
1 # sysctl -w vm.panic_on_oom=1 2 vm.panic_on_oom = 1 3 4 # sysctl -w kernel.panic=10 5 kernel.panic = 10 6 7 # echo "vm.panic_on_oom=1" >> /etc/sysctl.conf 8 # echo "kernel.panic=10" >> /etc/sysctl.conf
从上面的 oom_kill.c 代码里可以看到 oom_badness() 给每个进程打分，根据 points 的高低来决定杀哪个进程，这个 points 可以根据 adj 调节，root 权限的进程通常被认为很重要，不应该被轻易杀掉，所以打分的时候可以得到 3% 的优惠（adj -= 30; 分数越低越不容易被杀掉）。我们可以在用户空间通过操作每个进程的 oom_adj 内核参数来决定哪些进程不这么容易被 OOM killer 选中杀掉。比如，如果不想 MySQL 进程被轻易杀掉的话可以找到 MySQL 运行的进程号后，调整 oom_score_adj 为 -15（注意 points 越小越不容易被杀）：
1 # ps aux | grep mysqld 2 mysql 2196 1.6 2.1 623800 44876 ? Ssl 09:42 0:00 /usr/sbin/mysqld 3 4 # cat /proc/2196/oom_score_adj 5 0 6 # echo -15 > /proc/2196/oom_score_adj
当然，如果需要的话可以完全关闭 OOM killer（不推荐用在生产环境）：
1 # sysctl -w vm.overcommit_memory=2 2 3 # echo "vm.overcommit_memory=2" >> /etc/sysctl.conf