-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsetup.sh
More file actions
executable file
·58 lines (47 loc) · 1.97 KB
/
setup.sh
File metadata and controls
executable file
·58 lines (47 loc) · 1.97 KB
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
#!/bin/bash
# setup.sh - Script de configuração inicial do projeto AI Image Generation
# Created: September 3, 2025
set -e # Abort script at first error
# Cores para output
GREEN='\033[0;32m'
BLUE='\033[0;34m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m' # No Color
echo -e "${BLUE}=== AI Image Generation Service - Configuração Inicial ===${NC}"
# Verificar se python3 está instalado
if ! command -v python3 &> /dev/null; then
echo -e "${RED}Python 3 não encontrado. Por favor, instale o Python 3.8 ou superior.${NC}"
exit 1
fi
# Verificar versão do Python
python_version=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
echo -e "${GREEN}Python $python_version encontrado${NC}"
# Criar ambiente virtual
echo -e "${BLUE}Criando ambiente virtual...${NC}"
python3 -m venv venv
source venv/bin/activate
# Instalar dependências
echo -e "${BLUE}Instalando dependências...${NC}"
pip install --upgrade pip
pip install -r requirements.txt
# Criar diretórios necessários
echo -e "${BLUE}Configurando diretórios...${NC}"
mkdir -p output logs .cache
# Configurar arquivo .env
if [ ! -f .env ]; then
echo -e "${YELLOW}Criando arquivo .env a partir do modelo...${NC}"
cp .env.example .env
echo -e "${YELLOW}ATENÇÃO: Configure seu arquivo .env conforme necessário${NC}"
else
echo -e "${GREEN}Arquivo .env já existe${NC}"
fi
# Verificar suporte a GPU
echo -e "${BLUE}Verificando suporte a aceleração de hardware...${NC}"
python -c "import torch; print(f'PyTorch: {torch.__version__}')"
python -c "import torch; print(f'CUDA disponível: {torch.cuda.is_available()}')"
python -c "import torch; print(f'MPS disponível: {getattr(torch, \"backends\", None) and getattr(torch.backends, \"mps\", None) and getattr(torch.backends.mps, \"is_available\", lambda: False)()}')"
echo -e "${GREEN}✓ Configuração concluída!${NC}"
echo -e "${YELLOW}Para iniciar o servidor, execute:${NC}"
echo -e " source venv/bin/activate"
echo -e " python server.py"