services:
  nnue-pytorch:
    restart: no
    build:
      dockerfile: ./Dockerfile
    container_name: nnue-pytorch
    image: nnue-pytorch
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      - SSH_AUTH_SOCK=/tmp/ssh-agent
    volumes:
      - /dev/shm/:/dev/shm
      - ${SSH_AUTH_SOCK}:/tmp/ssh-agent