23 lines
530 B
YAML
23 lines
530 B
YAML
services:
|
|
testgpupodman_count:
|
|
image: docker.io/ubuntu:latest
|
|
command: ["nvidia-smi"]
|
|
profiles: [gpu]
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
test:
|
|
image: docker.io/ubuntu:latest
|
|
command: ["nvidia-smi"]
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: cdi
|
|
count: 1
|
|
device_ids: ['nvidia.com/gpu=all']
|
|
capabilities: [gpu] |