podman-compose/examples/nvidia-smi/docker-compose.yaml

12 lines
245 B
YAML
Raw Normal View History

services:
test:
image: nvidia/cuda:12.3.1-base-ubuntu20.04
command: nvidia-smi
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]