kind: pipeline
name: train
type: docker

concurrency:
  limit: 1

steps:
  - name: install
    image: nvcr.io/nvidia/pytorch:23.03-py3
    commands:
      - pip install torchsummary
      - ls ./