wisrovi/basi_gpu repository overview

services:
  gpu-api:
    image: wisrovi/basi_gpu
    ports:
      - "8001:8000"
     restart: unless-stopped
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]