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