mirror of
https://github.com/rishikanthc/Scriberr.git
synced 2026-03-03 02:17:01 +00:00
8f537548d4
Add support for NVIDIA RTX 50-series GPUs (Blackwell architecture) which require CUDA 12.8+ and PyTorch cu128 wheels due to the new sm_120 compute capability. Changes: - Add configurable PYTORCH_CUDA_VERSION environment variable to control PyTorch wheel version at runtime (cu126 for legacy, cu128 for Blackwell) - Update all model adapters to use dynamic CUDA version instead of hardcoded cu126 URLs - Update Dockerfile.cuda.12.9 for Blackwell with CUDA 12.9.1 base image, PYTORCH_CUDA_VERSION=cu128, and missing WHISPERX_ENV/yt-dlp - Update Dockerfile.cuda with explicit PYTORCH_CUDA_VERSION=cu126 - Add docker-compose.blackwell.yml for pre-built Blackwell image - Add docker-compose.build.blackwell.yml for local Blackwell builds - Add GPU compatibility documentation to README Fixes: rishikanthc/Scriberr#104
35 lines
1.0 KiB
YAML
35 lines
1.0 KiB
YAML
# Docker Compose for Scriberr with Blackwell GPU support (RTX 50-series)
|
|
# Uses pre-built image from GitHub Container Registry
|
|
# For legacy GPUs (GTX 10-series through RTX 40-series), use docker-compose.cuda.yml instead
|
|
version: "3.9"
|
|
services:
|
|
scriberr:
|
|
image: ghcr.io/rishikanthc/scriberr-cuda-blackwell:latest
|
|
ports:
|
|
- "8080:8080"
|
|
volumes:
|
|
- scriberr_data:/app/data
|
|
- env_data:/app/whisperx-env
|
|
restart: unless-stopped
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities:
|
|
- gpu
|
|
environment:
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
- PUID=${PUID:-1000}
|
|
- PGID=${PGID:-1000}
|
|
# Security: already set in container, but can be overridden
|
|
- APP_ENV=production
|
|
# CORS: comma-separated list of allowed origins for production
|
|
# - ALLOWED_ORIGINS=https://your-domain.com
|
|
|
|
volumes:
|
|
scriberr_data: {}
|
|
env_data: {}
|