Как указать ограничение графического процессора, процессора и памяти с помощью docker compose?
В этом случае я следую инструкциям в (https://nvidia.github.io/nvidia-container-runtime/).
Используя:
$ apt-get install nvidia-container-runtime
Я также проверяю это с помощью:
$ docker run -it --rm --gpus all ubuntu nvidia-smi
Примеры docker-compose.yaml
:
version: '2'
...
services:
...
my-service:
...
deploy:
resources:
reservations:
generic_resources:
- discrete_resource_spec:
kind: 'gpu'
value: 1
limits:
cpus: '4'
memory: 4096M
another-service:
...
deploy:
resources:
limits:
cpus: '0.001'
memory: 50M