Skip to content

Conversation

@loserbcc
Copy link

Summary

This PR adds a comprehensive Docker containerization solution for ACE-Step, optimized for modern NVIDIA GPUs and production deployment.

Features

🐳 Docker Infrastructure

  • Optimized Dockerfile (Dockerfile.scorpy) with CUDA 13.0 support for RTX 5090 and newer GPUs
  • Docker Compose configuration with full GPU support and health checks
  • Persistent volumes for checkpoints, outputs, and logs
  • CPU offload mode enabled (~0.4GB VRAM usage)

🛠️ Management Tools

  • Quick launcher (ace-docker) - Single command to start/stop/manage
  • Full management script (docker-manage.sh) - Complete lifecycle management
  • Helper scripts for different quality presets
  • Comprehensive documentation (DOCKER.md)

⚡ Performance Optimizations

  • BFloat16 precision for faster inference
  • Overlapped decode for better throughput
  • Tested on RTX 5090 with excellent performance

Usage

# Ultra-simple launch
./ace-docker

# Or use full management interface
./docker-manage.sh start   # Start container
./docker-manage.sh logs    # View logs
./docker-manage.sh status  # Check status
./docker-manage.sh rebuild # Full rebuild

Access the web interface at: http://localhost:7866

Files Added

  • .dockerignore - Build optimization
  • Dockerfile.scorpy - CUDA 13.0 optimized build
  • docker-compose.scorpy.yml - Container orchestration
  • docker-manage.sh - Management script
  • ace-docker - Quick launcher
  • DOCKER.md - Complete documentation
  • Helper scripts for quality presets

Testing

This setup has been tested and verified working on:

  • ✅ RTX 5090 with CUDA 13.0
  • ✅ Arch Linux (should work on any Linux with Docker)
  • ✅ Production deployment with persistent volumes

Notes

  • Uses .scorpy suffix to avoid conflicts with existing Docker files
  • Complements rather than replaces existing Docker setup
  • All volumes are persistent and survive container restarts
  • Includes health checks and proper logging

Documentation

Complete setup instructions, troubleshooting, and advanced usage examples are included in DOCKER.md.


🤖 Generated with Claude Code

This commit adds a complete Docker containerization solution for ACE-Step:

Features:
- Optimized Dockerfile for CUDA 13.0/RTX 5090
- Docker Compose configuration with GPU support
- Management script with easy commands (start/stop/logs/rebuild)
- Quick launcher script (ace-docker) for simple usage
- Comprehensive documentation (DOCKER.md)
- Helper scripts for different quality settings
- CPU offload enabled (~0.4GB VRAM usage)
- Persistent volumes for checkpoints, outputs, and logs

Quick start:
  ./ace-docker         # Start ACE-Step
  ./docker-manage.sh   # Full management interface

Access at: http://localhost:7866

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>
@tailoralm
Copy link

thank you so much! I have an RTX 5070 and you saved me

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants