Use dustynv/vllm base image for Jetson CUDA support
This commit is contained in:
18
Dockerfile
18
Dockerfile
@@ -1,11 +1,9 @@
|
|||||||
# OrpheusTail - Orpheus TTS Service for NVIDIA Jetson AGX Orin
|
# OrpheusTail - Orpheus TTS Service for NVIDIA Jetson AGX Orin
|
||||||
#
|
#
|
||||||
# Replaces VoiceTail (Bark) with Orpheus for better emotion control
|
# Uses dustynv's vLLM container which has proper Jetson CUDA support
|
||||||
# and voice cloning capabilities.
|
# Orpheus uses vLLM under the hood for fast inference
|
||||||
#
|
|
||||||
# Based on NVIDIA L4T PyTorch container optimized for Jetson
|
|
||||||
|
|
||||||
FROM dustynv/pytorch:2.1-r36.2.0
|
FROM dustynv/vllm:0.8.6-r36.4-cu128-24.04
|
||||||
|
|
||||||
# Set working directory
|
# Set working directory
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
@@ -20,13 +18,13 @@ RUN apt-get update && apt-get install -y \
|
|||||||
# Copy requirements first for better caching
|
# Copy requirements first for better caching
|
||||||
COPY requirements.txt /app/
|
COPY requirements.txt /app/
|
||||||
|
|
||||||
# Install Python dependencies
|
# Install Python dependencies (FastAPI, etc - but NOT torch/vllm)
|
||||||
# Note: torch and torchvision are already in the base image
|
|
||||||
RUN pip3 install --no-cache-dir -r requirements.txt
|
RUN pip3 install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
# Install orpheus-speech (uses vllm under the hood)
|
# Install orpheus-speech WITHOUT dependencies to avoid overwriting vllm/torch
|
||||||
# Note: vllm version compatibility may need adjustment
|
# Then install just the audio codec it needs (snac)
|
||||||
RUN pip3 install orpheus-speech
|
RUN pip3 install --no-cache-dir --no-deps orpheus-speech && \
|
||||||
|
pip3 install --no-cache-dir snac
|
||||||
|
|
||||||
# Copy application code
|
# Copy application code
|
||||||
COPY main.py /app/
|
COPY main.py /app/
|
||||||
|
|||||||
Reference in New Issue
Block a user