24 lines
381 B
YAML
24 lines
381 B
YAML
kind: pipeline
|
|
name: train
|
|
type: docker
|
|
|
|
concurrency:
|
|
limit: 1
|
|
|
|
steps:
|
|
- name: install
|
|
image: nvcr.io/nvidia/pytorch:23.03-py3
|
|
environment:
|
|
mount_command:
|
|
from_secret: mount_command
|
|
commands:
|
|
- python -m pip install --upgrade pip
|
|
- pip install torchsummary
|
|
- ls ./
|
|
- mkdir -p data
|
|
- bash mount_command
|
|
- ls data
|
|
|
|
|
|
|