This commit is contained in:
Malte Grosse 2023-11-28 22:48:30 +09:00
parent 76b687bb9c
commit 4603bb64a7
1 changed files with 11 additions and 1 deletions

View File

@ -12,6 +12,12 @@ steps:
- nvidia-smi
# environment:
# - NVIDIA_VISIBLE_DEVICES=all
backend_options:
kubernetes:
resources:
limits:
nvidia.com/mig-2g.20gb: 1
gpu:
image: oguzpastirmaci/gpu-burn:latest
# environment:
@ -20,7 +26,11 @@ steps:
- cd /app
- ./gpu_burn 120
- echo "burned. done"
backend_options:
kubernetes:
resources:
limits:
nvidia.com/mig-2g.20gb: 1
# 2nd:
# image: oguzpastirmaci/gpu-burn:latest
# environment: