File size: 1,491 Bytes
0113e69 35218f9 0113e69 35218f9 0113e69 35218f9 0113e69 aef4a65 0113e69 35218f9 0113e69 aef4a65 35218f9 aef4a65 0113e69 aef4a65 35218f9 aef4a65 35218f9 aef4a65 0113e69 |
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 |
# Nome do ambiente Conda que será criado
name: seedvr
# Canais de onde o Conda irá baixar os pacotes, em ordem de prioridade.
# pytorch e nvidia são essenciais para obter as versões corretas com suporte a GPU.
channels:
- pytorch
- nvidia
- defaults
# Lista de todas as dependências
dependencies:
# Pacotes principais instalados via Conda para máxima compatibilidade com a GPU
- python
- pytorch
- torchvision
- torchaudio
- pytorch-cuda # Garante que o PyTorch seja compatível com a CUDA da imagem Docker
- pip # Instala o gerenciador de pacotes pip dentro do ambiente
# Pacotes instalados via Pip
- pip:
# Dependências da API FastAPI
- fastapi
- uvicorn
- python-multipart
# Dependências do projeto SeedVR
- accelerate
- av
- beautifulsoup4
- controlnet_aux
- diffusers
- einops
- ftfy
- huggingface_hub
- iopath
- mediapy
- omegaconf
- opencv-python
- pandas
- pytorch-lightning
- pytorchvideo
- regex
- scikit-image
- timm
- tqdm
- transformers
- webdataset
- xformers
# NOTA IMPORTANTE:
# 'flash_attn' não está listado aqui de propósito.
# Sua instalação requer flags especiais para evitar erros de memória durante a construção.
# Por isso, ele é instalado separadamente com um comando RUN dedicado no Dockerfile. # A linha do flash_attn foi REMOVIDA daqui |