28 lines
825 B
Docker
28 lines
825 B
Docker
# Utilisation d'une image légère avec support GPU
|
|
FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
|
|
|
|
# Configuration environnement
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
WORKDIR /app
|
|
|
|
# Installation des dépendances système (Audio + Python)
|
|
RUN apt-get update && apt-get install -y \
|
|
python3.11 python3-pip ffmpeg libsndfile1 git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# 1. Installer d'abord PyTorch depuis la source NVIDIA/CUDA
|
|
RUN pip3 install --no-cache-dir \
|
|
torch torchvision torchaudio \
|
|
--index-url https://download.pytorch.org/whl/cu121
|
|
|
|
# 2. Installer le reste depuis le dépôt Python standard (PyPI)
|
|
RUN pip3 install --no-cache-dir \
|
|
fastapi uvicorn gradio
|
|
|
|
# Copie du code serveur
|
|
COPY server.py .
|
|
|
|
# Exposition du port pour la communication interne
|
|
EXPOSE 7860
|
|
|
|
CMD ["python3.11", "server.py"] |