1. 拉取MMPose代码
git clone https://github.com/open-mmlab/mmpose.git
2. 根据MMPose dockerfile构建一个docker image
cd mmpose
docker build -t mmpose docker/
3. 编写docker-compose.yml文件
version: '3.8'
services:
mmpose:
image: mmpose
container_name: mmpose_vitpose_container
pull_policy: never
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
shm_size: '8gb'
stdin_open: true
tty: true
volumes:
- ./mmpose_data:/mmpose/data
- ./mmpose:/mmpose
command: bash
4. 启动docker container
docker compose up -d # 启动,后台挂载
docker exec -it mmpose_vitpose_container /bin/bash # 用指定的container名启动
命令附录
| 作用 | 命令 |
|---|
| 后台启动container | docker compose up -d |
| 关闭container | docker compose down |
| 列出所有container | docker ps |
| 列出所有images | docker images |
| 根据dockerfile来构建Docker镜像 | docker build -t <镜像名称> <Dockerfile路径> |