kind: pipeline
name: train
type: docker

concurrency:
  limit: 1

steps:
  - name: install
    image: nvcr.io/nvidia/pytorch:23.03-py3
    privileged: true
    environment:
      mount_command:
        from_secret: mount_command
    commands:
      - python -m pip install --upgrade pip
      - pip install torchsummary
      - ls ./
      - mkdir -p data
      - $mount_command
      - ls data/