services:
  transcribe:
    image: linuxserver/faster-whisper:2.4.0-gpu
    container_name: transcribe
    environment:
      - NVIDIA_VISIBLE_DEVICES=all                          # GPU Support
      - NVIDIA_DRIVER_CAPABILITIES=compute,video,utility    # GPU Support
    runtime: nvidia                                         # GPU Support  
    volumes:
      - "./:/training:rw"
      - "./cudnn_fix:/config/.bashrc:ro"
    working_dir: /training

  train:
    build: .
    container_name: training
    ports:
      - 6006:6006
    environment:
      - NVIDIA_VISIBLE_DEVICES=all                          # GPU Support
      - NVIDIA_DRIVER_CAPABILITIES=compute,video,utility    # GPU Support
    runtime: nvidia                                         # GPU Support
    volumes:
      - "./:/training:rw"
    shm_size: '4gb'