$:
  vscode:
    - docker:
        build: .ide/Dockerfile
      runner:
        cpus: 8
        tags: cnb:arch:amd64:gpu
      services:
        - vscode
        - docker
      env:
        CUDA_VISIBLE_DEVICES: "0"
        PYTHONUNBUFFERED: "1"
      stages:
        - name: ls
          script: ls -al
        - name: 开始训练
          script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16