【ROS2】 Jazzy + YOLOv8 极简 Docker 实战:拒绝臃肿,代码热挂载!

发文时区: 上海

在机器人开发中,我们常面临一个两难选择:环境配置太麻烦,Docker 镜像又太大。特别是当我们需要结合 ROS2 Jazzy (Ubuntu 24.04)深度学习 (PyTorch/YOLO) 时,动辄 20GB+ 的镜像不仅构建慢,还难以维护。

今天分享一套我亲测有效的**“极简主义”**方案:打造一个仅包含基础运行时的轻量级 Docker 镜像,将业务代码完全剥离到宿主机。

核心优势:

  • ⚡ 构建极快:镜像体积控制在 6GB 左右(含 PyTorch)。
  • 🔥 代码热更:修改 Python 代码无需重新 build,保存即生效。
  • 🛡️ 环境隔离:完美解决 Ubuntu 24.04 的 --break-system-packages 权限问题。

🛠️ 第一步:编写极简 Dockerfile

这个 Dockerfile 只做三件事:提供 ROS2 环境、强制升级 pip、安装 PyTorch。

新建 Dockerfile,复制以下内容:

# 1. 基础镜像:使用 DaoCloud 加速的 ROS2 Jazzy Desktop Full (Ubuntu 24.04)
FROM docker.m.daocloud.io/osrf/ros:jazzy-desktop-full

# 2. 设置非交互式环境,防止 apt 弹窗卡死
ENV DEBIAN_FRONTEND=noninteractive

# 3. 安装系统级核心依赖
# 【注意】这里没有安装 python3-numpy,避免与后续 pip 安装的版本冲突
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3-opencv \
    python3-pip \
    git \
    && rm -rf /var/lib/apt/lists/*

# 4. 配置清华源 & 强制升级 Pip
# 使用 --ignore-installed 强制覆盖系统自带的 pip,解决部分系统 pip 权限问题
RUN pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip3 install --break-system-packages --ignore-installed pip

# 5. 【耗时层】安装 PyTorch (CUDA 11.8)
# 这一层独立出来,利用 Docker 缓存机制,避免每次修改代码都重下 2GB+ 的 PyTorch
RUN pip3 install --break-system-packages \
    torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

# 6. 【快速层】安装 Ultralytics, OpenCV 和 Numpy
# 【关键修复】加入 --ignore-installed 彻底解决 numpy RECORD file 丢失导致的卸载报错
RUN pip3 install --break-system-packages --ignore-installed \
    ultralytics \
    opencv-python-headless \
    numpy

# 7. 设置工作目录
WORKDIR /root/workspace

# 8. 默认启动 bash
CMD ["/bin/bash"]

💡 关键点解析:

  • DaoCloud 镜像:利用国内加速源,避免拉取官方镜像超时。
  • --ignore-installed pip:这是解决 Ubuntu 24.04 下 RECORD file not found 报错的核心。它会强制覆盖系统自带的旧版 pip,无需手动卸载。
  • CUDA 11.8:兼容性最好,适配大多数 NVIDIA 显卡(包括 RTX A4500/4090 等)。

🏗️ 第二步:构建与启动

1. 构建镜像
在 Dockerfile 所在目录执行:

docker build -t jazzy_ai_base .

2. 启动容器(核心:挂载代码 + GPU)
假设你的 ROS2 工作空间在宿主机的 /home/liurui/ros2_ws,使用以下命令启动:

docker run -it --rm \
  --name jazzy_dev \
  --gpus all \
  --network host \
  --privileged \
  -e DISPLAY=$DISPLAY \
  -v /tmp/.X11-unix:/tmp/.X11-unix \
  -v /home/liurui/ros2_ws/src:/root/ros2_ws/src \
  jazzy_ai_base

参数说明:

  • --gpus all必须。启用 NVIDIA 显卡加速,YOLO 推理才能实时。
  • -v /home/...:将宿主机代码挂载到容器内,实现代码同步。

🐍 第三步:在容器内跑代码

进入容器后,执行以下命令:

# 1. 创建工作空间目录(如果不存在)
mkdir -p /root/ros2_ws/src

# 2. 进入工作空间并编译(首次运行需要)
cd /root/ros2_ws
colcon build --symlink-install
source install/setup.bash

# 3. 验证 PyTorch GPU
python3 -c "import torch; print(torch.cuda.is_available())"
# 输出 True 说明 GPU 配置成功

# 4. 运行你的节点
ros2 run my_robot_controller circle_driver

此时,你在宿主机修改 circle_driver.py,容器内重启节点即可生效,无需重新构建镜像!


⚠️ 避坑指南 (Troubleshooting)
  • 关于 RECORD file not found 错误

    • 原因:Ubuntu 24.04 系统自带的 pip 由 apt 管理,缺少卸载元数据。
    • 解决:Dockerfile 中必须使用 --ignore-installed pip 强制覆盖安装,而不是 --upgrade
  • 关于 Gazebo 黑屏或无法显示

    • 解决:在宿主机终端执行 xhost +local:docker,授予容器 GUI 权限。
  • 关于显卡驱动版本

    • 如果你的驱动较新(支持 CUDA 12),可以将 Dockerfile 中的 cu118 改为 cu121,以获得更好的性能。

📝 总结

这套方案将 “环境”“代码” 彻底解耦。

  • 环境坏了? -> 删掉容器,docker run 重来,1秒恢复。
  • 代码改了? -> 宿主机保存,容器内重启,0秒同步。
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐