docker-gpu-tests/docker-compose.yml

44 lines
1.1 KiB
YAML

version: "3.8"
services:
test1:
image: stress-test
build:
context: ./
dockerfile: stressDockerfile
environment:
- NVIDIA_VISIBLE_DEVICES=all
working_dir: /gpu-burn
command:
[
"./gpu_burn",
"120"
]
deploy:
resources:
reservations:
devices:
- 'driver': 'nvidia'
'count': 1
'capabilities': ['gpu', 'utility', 'compute']
test2:
image: stress-test
build:
context: ./
dockerfile: stressDockerfile
environment:
- NVIDIA_VISIBLE_DEVICES=all
working_dir: /gpu-burn
command:
[
"./gpu_burn",
"120"
]
deploy:
resources:
reservations:
devices:
- 'driver': 'nvidia'
'count': 1
'capabilities': ['gpu', 'utility', 'compute']