docker-compose.yml

services:
  make_env:
    image: safe-make-env:latest
    build: .
    command: bash make_custom_env.sh
    volumes:
      - /disk1/kirill/custom-env:/app/custom-env
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: [ '0','1','2','3']
              capabilities: [ gpu ]