84 lines
2.2 KiB
Docker
84 lines
2.2 KiB
Docker
|
# AudioCraft Studio Dockerfile for RunPod
|
||
|
|
# Optimized for NVIDIA RTX 4090 (24GB VRAM)
|
||
|
|
|
||
|
|
FROM nvidia/cuda:12.1-cudnn8-runtime-ubuntu22.04
|
||
|
|
|
||
|
|
# Set environment variables
|
||
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
||
|
|
ENV PYTHONUNBUFFERED=1
|
||
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
||
|
|
ENV PIP_NO_CACHE_DIR=1
|
||
|
|
ENV PIP_DISABLE_PIP_VERSION_CHECK=1
|
||
|
|
|
||
|
|
# CUDA settings
|
||
|
|
ENV CUDA_HOME=/usr/local/cuda
|
||
|
|
ENV PATH="${CUDA_HOME}/bin:${PATH}"
|
||
|
|
ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
|
||
|
|
|
||
|
|
# AudioCraft settings
|
||
|
|
ENV AUDIOCRAFT_OUTPUT_DIR=/workspace/outputs
|
||
|
|
ENV AUDIOCRAFT_DATA_DIR=/workspace/data
|
||
|
|
ENV AUDIOCRAFT_MODEL_CACHE=/workspace/models
|
||
|
|
ENV AUDIOCRAFT_HOST=0.0.0.0
|
||
|
|
ENV AUDIOCRAFT_GRADIO_PORT=7860
|
||
|
|
ENV AUDIOCRAFT_API_PORT=8000
|
||
|
|
|
||
|
|
# Install system dependencies
|
||
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||
|
|
git \
|
||
|
|
curl \
|
||
|
|
wget \
|
||
|
|
ffmpeg \
|
||
|
|
libsndfile1 \
|
||
|
|
libsox-dev \
|
||
|
|
sox \
|
||
|
|
build-essential \
|
||
|
|
python3.10 \
|
||
|
|
python3.10-venv \
|
||
|
|
python3.10-dev \
|
||
|
|
python3-pip \
|
||
|
|
&& rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
# Set Python 3.10 as default
|
||
|
|
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 \
|
||
|
|
&& update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1
|
||
|
|
|
||
|
|
# Upgrade pip
|
||
|
|
RUN pip install --upgrade pip setuptools wheel
|
||
|
|
|
||
|
|
# Create workspace directory
|
||
|
|
WORKDIR /workspace
|
||
|
|
|
||
|
|
# Create necessary directories
|
||
|
|
RUN mkdir -p /workspace/outputs /workspace/data /workspace/models /workspace/app
|
||
|
|
|
||
|
|
# Copy requirements first for caching
|
||
|
|
COPY requirements.txt /workspace/app/
|
||
|
|
WORKDIR /workspace/app
|
||
|
|
|
||
|
|
# Install PyTorch with CUDA support
|
||
|
|
RUN pip install torch==2.1.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
|
||
|
|
|
||
|
|
# Install other requirements
|
||
|
|
RUN pip install -r requirements.txt
|
||
|
|
|
||
|
|
# Install AudioCraft from source for latest features
|
||
|
|
RUN pip install git+https://github.com/facebookresearch/audiocraft.git
|
||
|
|
|
||
|
|
# Copy application code
|
||
|
|
COPY . /workspace/app/
|
||
|
|
|
||
|
|
# Create non-root user for security (optional, RunPod often uses root)
|
||
|
|
# RUN useradd -m -u 1000 audiocraft && chown -R audiocraft:audiocraft /workspace
|
||
|
|
# USER audiocraft
|
||
|
|
|
||
|
|
# Expose ports
|
||
|
|
EXPOSE 7860 8000
|
||
|
|
|
||
|
|
# Health check
|
||
|
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
|
||
|
|
CMD curl -f http://localhost:7860/ || exit 1
|
||
|
|
|
||
|
|
# Default command
|
||
|
|
CMD ["python", "main.py"]
|