在 Ubuntu 中配置开机自启动脚本并激活 Anaconda 环境

在 Ubuntu 中配置开机自启动脚本并激活 Anaconda 环境

在 Ubuntu 系统中，配置开机自启动脚本并激活 Anaconda 环境是许多开发者和数据科学家的常见需求。本文将详细介绍如何通过 systemd 服务单元 或 init.d 脚本 实现这一目标，并解决在非交互式 shell 中激活 Anaconda 环境时可能遇到的问题。

1. 使用 Systemd 服务单元（推荐）

1.1 创建服务单元文件

首先，创建一个 systemd 服务单元文件：

sudo nano /etc/systemd/system/your-service-name.service

1.2 配置服务内容

在文件中填写以下内容（根据实际情况调整）：

[Unit]
Description=启动我的服务并激活 Anaconda 环境
After=network.target

[Service]
# 指定运行脚本的用户
User=your-username
# 启动脚本
ExecStart=/usr/bin/bash -c 'source /home/your-username/anaconda3/etc/profile.d/conda.sh && conda activate your-env-name && /path/to/your_script.py'
# 失败时重启服务
Restart=on-failure

[Install]
WantedBy=multi-user.target

1.3 参数说明

• User=your-username：替换为你的用户名。
• ExecStart：
  • source /home/your-username/anaconda3/etc/profile.d/conda.sh：手动加载 Conda 的环境变量。
  • conda activate your-env-name：激活指定的 Anaconda 环境。
  • /path/to/your_script.py：替换为你的脚本路径。

1.4 重新加载并启用服务

sudo systemctl daemon-reload
sudo systemctl enable your-service-name.service
sudo systemctl start your-service-name.service

1.5 验证服务

systemctl status your-service-name.service
journalctl -u your-service-name.service --since "1 minute ago"

2. 使用 init.d 脚本（兼容旧系统）

2.1 创建 init.d 脚本

sudo nano /etc/init.d/your-script-name

2.2 添加内容

#!/bin/sh
### BEGIN INIT INFO
# Provides:          your-script-name
# Required-Start:    $remote_fs $syslog
# Required-Stop:     $remote_fs $syslog
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: 启动并激活 Anaconda 环境的脚本
### END INIT INFO

case "$1" in
  start)
    echo "启动服务..."
    /bin/bash -c 'source /home/your-username/anaconda3/etc/profile.d/conda.sh && conda activate your-env-name && /path/to/your_script.py &' &
    ;;
  stop)
    echo "停止服务..."
    pkill -f your_script.py
    ;;
  restart)
    $0 stop
    $0 start
    ;;
  *)
    echo "使用: {start|stop|restart}"
    exit 1
    ;;
esac

exit 0

2.3 设置权限并启用服务

sudo chmod +x /etc/init.d/your-script-name
sudo update-rc.d your-script-name defaults

2.4 启动服务并验证

sudo /etc/init.d/your-script-name start
tail -f /var/log/syslog  # 查看系统日志

3. 关键注意事项

3.1 路径必须绝对

• Anaconda 安装路径通常是用户的家目录下的 anaconda3 或 miniconda3（例如 /home/your-username/anaconda3）。
• 脚本路径需使用绝对路径（如 /home/your-username/script.py）。

3.2 环境变量问题

在非交互式 shell 中，conda activate 不会自动加载环境变量，因此必须手动 source conda.sh。例如：

#!/bin/bash
source /home/your-username/anaconda3/etc/profile.d/conda.sh
conda activate your-env-name
python your_script.py

3.3 权限问题

• 确保服务以正确用户身份运行（通过 User=your-username 或脚本的 sudo -u）。
• 确保脚本有执行权限：

  chmod +x /path/to/your_script.py
  

3.4 日志调试

向脚本添加日志输出，例如：

echo "正在激活环境..." >> /tmp/my_script.log
echo "当前环境变量：" >> /tmp/my_script.log
env >> /tmp/my_script.log

4. 示例：验证 Anaconda 环境激活的脚本

4.1 创建测试脚本 test_script.sh

#!/bin/bash
source /home/your-username/anaconda3/etc/profile.d/conda.sh
conda activate your-env-name
echo "当前激活的环境：" $(conda info --envs | grep "*" | awk '{print $NF}') >> /tmp/test.log
echo "Python 版本：" >> /tmp/test.log
python --version >> /tmp/test.log

4.2 修改 service 文件或 init.d 脚本，指向此测试脚本。

4.3 验证输出

tail -f /tmp/test.log

5. 总结

通过以上步骤，你可以在 Ubuntu 的开机自启动脚本中正确激活并使用 Anaconda 环境。如果遇到权限或路径问题，请仔细检查日志并确保所有路径均为绝对路径。希望本文能帮助你轻松实现这一需求！