26 lines
684 B
Text
26 lines
684 B
Text
|
FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04
|
||
|
|
|
||
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
||
|
|
ENV PYTHONUNBUFFERED=1
|
||
|
|
|
||
|
|
RUN apt-get update && apt-get install -y \
|
||
|
|
python3.11 python3.11-venv python3-pip \
|
||
|
|
ffmpeg \
|
||
|
|
&& rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
COPY requirements.txt .
|
||
|
|
RUN pip3 install --no-cache-dir -r requirements.txt
|
||
|
|
RUN pip3 install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu121
|
||
|
|
|
||
|
|
RUN python3 -c "from faster_whisper import WhisperModel; WhisperModel('large-v3-turbo', device='cpu', compute_type='int8')"
|
||
|
|
|
||
|
|
COPY main.py .
|
||
|
|
|
||
|
|
EXPOSE 8000
|
||
|
|
|
||
|
|
CMD ["python3", "main.py"]
|