docker构建及算法镜像启动流程
1. 构建base镜像
解压docker.zip。
进入文件夹“step1_base”中,控制台输入指令
make build
会自动构建基础镜像yolobase_gpu:v1.0
2. 构建算法镜像
①将weights文件夹中的yolov8l.pt复制到model-yolov8/ultralytics-main/weights目录下
②进入算法代码存储路径model-yolov8/ultralytics-main,控制台输入指令
make build
会自动构建算法镜像yolov8_det:v1.0
3. 启动算法镜像
依旧在model-yolov8/ultralytics-main目录,控制台输入指令
docker-compose up -d
即可完成算法镜像的后台启动。
如果需要追踪算法镜像运行状态,可在启动后通过指令
docker-compose logs -f
4. docker-compose.yml注释
A对应的两个值都是flask服务在docker内部的端口号,因此修改时需要同步修改
B对应的是docker中特定端口号在宿主机的映射
其中:
必须要将授权文件映射到算法镜像的/usr/local/models目录下,授权文件的名称可以任意,只需要保证.lc后缀即可
5. 指令启动、非docker-compose启动
如果不想使用docker-compose启动镜像,启动指令如下
docker run --name yolo_detection --privileged --env NVIDIA_DRIVER_CAPABILITIES=compute,utility --env NVIDIA_VISIBLE_DEVICES=all --env FLASK_PORT=18089 --device /dev/mem:/dev/mem -v /dev/license.lc:/usr/local/models/license.lc -p 18089:18089 yolov8_det:v1.0 /bin/bash
6. docker-compose安装方法
apt install docker-compose
中途弹出选项,填y
注意:如果出现与docker有关的问题,可以尝试更改docker版本,降低版本,步骤如下:
卸载当前安装的 Docker:
sudo apt-get remove docker docker-engine docker.io containerd runc
更新 apt 软件包缓存:
sudo apt-get update
安装 Docker 20.10.21 版本:
sudo apt-get install docker-ce=5:20.10.21~3-0~ubuntu-focal docker-ce-cli=5:20.10.21~3-0~ubuntu-focal containerd.io
安装 Docker 时如果提示是否要添加 Docker GPG 密钥,请选择“是”或“是(Y)”。安装完成后,可以运行 `docker --version` 命令来验证 Docker 是否已正确安装,并显示正确的 Docker 版本号。