File size: 1,819 Bytes
8fb6272 b3cb0d8 8fb6272 b3cb0d8 8fb6272 2dfae6f 8fb6272 2dfae6f 8fb6272 2dfae6f 8fb6272 b3cb0d8 8fb6272 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
# Configurar ambiente não interativo
ENV DEBIAN_FRONTEND=noninteractive
WORKDIR /code
# Instalar Python e dependências do sistema
RUN apt-get update && apt-get install -y \
python3.10 \
python3-pip \
python3.10-venv \
libgl1-mesa-glx \
libglib2.0-0 \
ffmpeg \
git \
git-lfs \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
# Inicializar Git LFS
RUN git lfs install
# Copiar requirements primeiro para aproveitar cache
COPY requirements.txt .
# Configurar ambiente Python
ENV VIRTUAL_ENV=/opt/venv
RUN python3.10 -m venv $VIRTUAL_ENV
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
# Instalar dependências Python otimizadas para GPU
RUN pip install --no-cache-dir -U pip setuptools wheel && \
pip install --no-cache-dir torch torchvision --extra-index-url https://download.pytorch.org/whl/cu121 && \
pip install --no-cache-dir -r requirements.txt && \
pip install --no-cache-dir --upgrade "tokenizers>=0.15.0" && \
pip install --no-cache-dir --upgrade "git+https://github.com/huggingface/transformers.git"
# Criar diretório de vídeos
RUN mkdir -p /code/videos && chmod -R 777 /code/videos
# Configurar variáveis de ambiente
ENV HOST=0.0.0.0 \
PORT=7860 \
PYTHONUNBUFFERED=1 \
PYTHONPATH=/code \
TRANSFORMERS_CACHE=/tmp/huggingface \
TORCH_HOME=/tmp/torch \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_SERVER_PORT=7860 \
SYSTEM=spaces \
CUDA_VISIBLE_DEVICES=0 \
HUGGINGFACE_HUB_CACHE=/tmp/huggingface \
HF_HOME=/tmp/huggingface \
TORCH_CUDA_ARCH_LIST="7.5" \
MAX_WORKERS=2 \
TRANSFORMERS_OFFLINE=0 \
TRANSFORMERS_VERBOSITY=info
# Copiar arquivos do projeto
COPY . .
# Expor porta
EXPOSE 7860
# Comando para iniciar a aplicação
CMD ["python3", "app.py"]
|