今日提醒:优先选择采用docker部署,不建议采用本方案!!!

0. 应用背景

在一些AI项目中,往往通过接口对外提供服务,常用Docker来部署。在一些需要CUDA的场景,尽管有nvidia-docker容器可以使用,但配置过程相当繁琐。
可替换的方案(前提是已安装nvidia驱动、cuda+cudnn、conda并创建了运行环境):

  • 采用conda对服务的运行环境进行隔离
  • 布置开机自启动任务来后台执行代码
    接下来,将重点介绍开机自启动服务,以一个Python项目为例。

1. 查看启动脚本,其中就有我们需要的 rc.local.service

ls /lib/systemd/system

2. 打开rc-local.service脚本

可采用vi打开:

sudo vi /lib/systemd/system/rc-local.service

可以看到以下内容:

#  This file is part of systemd.
#
#  systemd is free software; you can redistribute it and/or modify it
#  under the terms of the GNU Lesser General Public License as published by
#  the Free Software Foundation; either version 2.1 of the License, or
#  (at your option) any later version.

# This unit gets pulled automatically into multi-user.target by
# systemd-rc-local-generator if /etc/rc.local is executable.
[Unit]
Description=/etc/rc.local Compatibility
ConditionFileIsExecutable=/etc/rc.local
After=network.target

[Service]
Type=forking
ExecStart=/etc/rc.local start
TimeoutSec=0
RemainAfterExit=yes

一般正常的启动文件主要分成三部分

[Unit] 段: 启动顺序与依赖关系
[Service] 段: 启动行为,如何启动,启动类型
[Install] 段: 定义如何安装这个配置文件,即怎样做到开机启动

可以看出,/etc/rc.local 的启动顺序是在网络后面,但是显然它少了 Install 段,也就没有定义如何做到开机启动,所以显然这样配置是无效的。 因此我们就需要在后面帮他加上 [Install] 段:

[Install]
WantedBy=multi-user.target  
Alias=rc-local.service

添加完成之后,如下图所示:

#  This file is part of systemd.
#
#  systemd is free software; you can redistribute it and/or modify it
#  under the terms of the GNU Lesser General Public License as published by
#  the Free Software Foundation; either version 2.1 of the License, or
#  (at your option) any later version.

# This unit gets pulled automatically into multi-user.target by
# systemd-rc-local-generator if /etc/rc.local is executable.
[Unit]
Description=/etc/rc.local Compatibility
ConditionFileIsExecutable=/etc/rc.local
After=network.target

[Service]
Type=forking
ExecStart=/etc/rc.local start
TimeoutSec=0
RemainAfterExit=yes

[Install]
WantedBy=multi-user.target  
Alias=rc-local.service

PS:添加了[Install]内容后,下面两行的WantedBy和Alias两个英文跟上面的都是绿色的,要是绿色才有用。

3. 编辑/etc/rc.local文件

进入etc文件夹,查看有没有 /etc/rc.local 这个文件,没有则自己创建一个。

sudo vi  /etc/rc.local

然后,把需要启动脚本写入 /etc/rc.local,此处以一个Python项目代码为例。

#!/bin/bash
source activate my_conda_env
cd /home/user/projects/my_python_project/
python3 run.py&
exit 0

注意:此处“#!/bin/sh”修改为“#!/bin/bash”,以执行source指令来激活conda运行环境。【此行为必须的!】
执行程序后必须加“&”!
以上过程依次进行了激活项目运行的conda环境、进入项目文件夹、执行运行程序。
然后给rc.local添加权限:

sudo chmod +x /etc/rc.local

4. 创建软链接

systemd 默认读取 /etc/systemd/system 下的配置文件, 所以还需要在 /etc/systemd/system 目录下创建软链接:

sudo ln -s /lib/systemd/system/rc.local.service /etc/systemd/system/

5. 验证是否成功

重启,在终端输入:

service service-name status

查看服务是否正在运行。
如果在 /etc/rc.local 中添加的是 ./xxx.sh 这种类型,或者是一个可执行程序(如本案中run.py),要在末尾加上“&”, 不然重启ubuntu的时候会卡在启动界面进不去系统。
万一如此,以下链接参考救砖。
致谢:Ubuntu20.04 开机自启动程序

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐