41 lines
897 B
YAML
41 lines
897 B
YAML
steps:
|
|
first-job:
|
|
image: busybox
|
|
commands:
|
|
- echo "ci working................. "
|
|
# cpu:
|
|
# image: progrium/stress:latest
|
|
# commands: /usr/bin/stress --cpu 20 --io 1 --vm 2 --vm-bytes 128M --timeout 90s
|
|
nvidia-test:
|
|
image: nvidia/cuda:11.6.2-base-ubuntu20.04
|
|
commands:
|
|
- nvidia-smi
|
|
backend_options:
|
|
kubernetes:
|
|
resources:
|
|
limits:
|
|
nvidia.com/mig-2g.20gb: 1
|
|
|
|
gpu:
|
|
image: oguzpastirmaci/gpu-burn:latest
|
|
# environment:
|
|
# - NVIDIA_VISIBLE_DEVICES=all
|
|
commands:
|
|
- cd /app
|
|
- ./gpu_burn 120
|
|
- echo "burned. done"
|
|
backend_options:
|
|
kubernetes:
|
|
resources:
|
|
requests:
|
|
limits:
|
|
nvidia.com/mig-2g.20gb: 1
|
|
# 2nd:
|
|
# image: oguzpastirmaci/gpu-burn:latest
|
|
# environment:
|
|
# - NVIDIA_VISIBLE_DEVICES=all
|
|
# commands:
|
|
# - ./gpu_burn 60
|
|
|
|
|