docker compose使用

docker compose 是docker官方编排项目之一，负责快速在集群中部署分布式应用；创建docker compose集群首先要编写好docker-compose.yaml 文件，也可以搭配docker-entrypoint.sh使用。

1. docker-compose.yaml 文件

version: '3.9'

services:
  dl-train:
    image: pytorch2cu11.7:20230921
    ports:
      - "8811:8811"
    volumes:
      - "/home/cai/czw/workspace/models:/workspace/models"
      - "/home/cai/czw/workspace/src:/workspace/src"
    networks:
      - sdnet
    restart: always
    entrypoint: ["sh","/workspace/src/jupyter-lab/docker-entrypoint.sh"]
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ['0']
              capabilities: [gpu]
networks:
  sdnet:
    ipam:
      driver: default
      config:
        - subnet: 192.168.12.0/24

2. docker-entrypoint.sh 文件

#!/bin/bash
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install --upgrade jupyterlab
# jupyter labextension install @jupyter-widgets/jupyterlab-manager
jupyter-lab --no-browser --allow-root --port=8811 --ip=0.0.0.0 --NotebookApp.token=123456

3.使用docker compose up -d 后台运行即可

docker compose down 停止和删除容器，网络