272 lines
8.4 KiB
Bash
Executable File
272 lines
8.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# 인페인팅 서버 자동 설치 스크립트
|
|
# 플랫폼 자동 감지 및 최적 설치 방법 선택
|
|
# Usage: ./install.sh
|
|
|
|
set -e
|
|
|
|
# 색상 코드
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
CYAN='\033[0;36m'
|
|
MAGENTA='\033[0;35m'
|
|
NC='\033[0m' # No Color
|
|
|
|
# 로고 출력
|
|
print_logo() {
|
|
echo -e "${CYAN}"
|
|
echo " ██║ ███╗ ██╗██████╗ █████╗ ██╗███╗ ██╗████████╗"
|
|
echo " ██║ ████╗ ██║██╔══██╗██╔══██╗██║████╗ ██║╚══██╔══╝"
|
|
echo " ██║ ██╔██╗ ██║██████╔╝███████║██║██╔██╗ ██║ ██║ "
|
|
echo " ██║ ██║╚██╗██║██╔═══╝ ██╔══██║██║██║╚██╗██║ ██║ "
|
|
echo " ██║ ██║ ╚████║██║ ██║ ██║██║██║ ╚████║ ██║ "
|
|
echo " ╚═╝ ╚═╝ ╚═══╝╚═╝ ╚═╝ ╚═╝╚═╝╚═╝ ╚═══╝ ╚═╝ "
|
|
echo ""
|
|
echo " 🎨 AI 인페인팅 서버 - 자동 설치 도구"
|
|
echo -e "${NC}"
|
|
}
|
|
|
|
# 로그 함수들
|
|
log_info() {
|
|
echo -e "${BLUE}[INFO]${NC} $1"
|
|
}
|
|
|
|
log_success() {
|
|
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
|
}
|
|
|
|
log_warning() {
|
|
echo -e "${YELLOW}[WARNING]${NC} $1"
|
|
}
|
|
|
|
log_error() {
|
|
echo -e "${RED}[ERROR]${NC} $1"
|
|
}
|
|
|
|
log_step() {
|
|
echo -e "${CYAN}[STEP]${NC} $1"
|
|
}
|
|
|
|
log_platform() {
|
|
echo -e "${MAGENTA}[PLATFORM]${NC} $1"
|
|
}
|
|
|
|
# 플랫폼 감지
|
|
detect_platform() {
|
|
local arch=$(uname -m)
|
|
local os_info=$(uname -a)
|
|
|
|
log_step "플랫폼 감지 중..."
|
|
|
|
if [[ "$arch" == "aarch64" ]] && echo "$os_info" | grep -q "tegra"; then
|
|
PLATFORM="jetson"
|
|
PLATFORM_NAME="NVIDIA Jetson Xavier"
|
|
INSTALL_SCRIPT="setup_jetson.sh"
|
|
log_platform "🚀 $PLATFORM_NAME 감지됨"
|
|
elif [[ "$arch" == "x86_64" ]] || [[ "$arch" == "amd64" ]]; then
|
|
PLATFORM="x86"
|
|
PLATFORM_NAME="x86-64 Desktop"
|
|
INSTALL_SCRIPT="setup_x86.sh"
|
|
log_platform "🖥️ $PLATFORM_NAME 감지됨"
|
|
elif [[ "$arch" == "aarch64" ]]; then
|
|
PLATFORM="arm64"
|
|
PLATFORM_NAME="ARM64 (Generic)"
|
|
INSTALL_SCRIPT="setup_jetson.sh"
|
|
log_platform "💻 $PLATFORM_NAME 감지됨 (Jetson 설치 방법 사용)"
|
|
else
|
|
log_error "지원하지 않는 아키텍처: $arch"
|
|
echo "지원 플랫폼:"
|
|
echo " - x86-64 (Intel/AMD)"
|
|
echo " - ARM64 (NVIDIA Jetson Xavier)"
|
|
exit 1
|
|
fi
|
|
|
|
echo "감지된 정보:"
|
|
echo " 아키텍처: $arch"
|
|
echo " 플랫폼: $PLATFORM_NAME"
|
|
echo " 설치 스크립트: $INSTALL_SCRIPT"
|
|
echo ""
|
|
}
|
|
|
|
# GPU 정보 표시
|
|
show_gpu_info() {
|
|
log_step "GPU 정보 확인 중..."
|
|
|
|
if command -v nvidia-smi >/dev/null 2>&1; then
|
|
echo "=== NVIDIA GPU 정보 ==="
|
|
nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader 2>/dev/null | while read line; do
|
|
echo " GPU: $line"
|
|
done
|
|
echo ""
|
|
else
|
|
log_warning "NVIDIA GPU 또는 드라이버가 설치되지 않았습니다."
|
|
echo ""
|
|
fi
|
|
}
|
|
|
|
# 시스템 요구사항 확인
|
|
check_requirements() {
|
|
log_step "시스템 요구사항 확인 중..."
|
|
|
|
local errors=0
|
|
|
|
# Python 3.8+ 확인
|
|
if command -v python3 >/dev/null 2>&1; then
|
|
local python_version=$(python3 --version 2>&1 | awk '{print $2}')
|
|
if python3 -c "import sys; exit(0 if sys.version_info >= (3, 8) else 1)" 2>/dev/null; then
|
|
log_success "Python $python_version 확인됨"
|
|
else
|
|
log_error "Python 3.8 이상이 필요합니다. 현재: $python_version"
|
|
errors=$((errors + 1))
|
|
fi
|
|
else
|
|
log_error "Python3가 설치되지 않았습니다."
|
|
errors=$((errors + 1))
|
|
fi
|
|
|
|
# Git 확인
|
|
if command -v git >/dev/null 2>&1; then
|
|
log_success "Git 확인됨"
|
|
else
|
|
log_warning "Git이 설치되지 않았습니다. 설치 과정에서 자동으로 설치됩니다."
|
|
fi
|
|
|
|
# 메모리 확인
|
|
if [ "$PLATFORM" = "jetson" ]; then
|
|
local total_mem=$(free -g | grep '^Mem:' | awk '{print $2}')
|
|
if [ "$total_mem" -ge 16 ]; then
|
|
log_success "메모리 ${total_mem}GB 확인됨 (Jetson 권장: 16GB+)"
|
|
else
|
|
log_warning "메모리 ${total_mem}GB (권장: 16GB+)"
|
|
fi
|
|
else
|
|
local total_mem=$(free -g | grep '^Mem:' | awk '{print $2}')
|
|
if [ "$total_mem" -ge 8 ]; then
|
|
log_success "메모리 ${total_mem}GB 확인됨"
|
|
else
|
|
log_warning "메모리 ${total_mem}GB (권장: 8GB+)"
|
|
fi
|
|
fi
|
|
|
|
# 디스크 공간 확인
|
|
local free_space=$(df -BG . | tail -1 | awk '{print $4}' | sed 's/G//')
|
|
if [ "$free_space" -ge 10 ]; then
|
|
log_success "디스크 여유 공간 ${free_space}GB 확인됨"
|
|
else
|
|
log_warning "디스크 여유 공간 ${free_space}GB (권장: 10GB+)"
|
|
fi
|
|
|
|
if [ $errors -gt 0 ]; then
|
|
log_error "필수 요구사항이 충족되지 않았습니다. 위 문제를 해결한 후 다시 시도하세요."
|
|
exit 1
|
|
fi
|
|
|
|
echo ""
|
|
}
|
|
|
|
# 설치 옵션 표시
|
|
show_installation_options() {
|
|
echo "=== 설치 옵션 ==="
|
|
echo ""
|
|
|
|
if [ "$PLATFORM" = "jetson" ]; then
|
|
echo "🚀 Jetson Xavier 최적화 설정:"
|
|
echo " - ONNX Runtime GPU (ARM64 특화)"
|
|
echo " - 32GB 통합 메모리 최적화"
|
|
echo " - CUDA + TensorRT 가속"
|
|
echo " - 세션 풀: LAMA(4), MIGAN(4), REMBG(3)"
|
|
echo " - 워커: 8개 (자동 조정)"
|
|
else
|
|
echo "🖥️ x86-64 데스크톱 설정:"
|
|
echo " - ONNX Runtime GPU (x86-64)"
|
|
echo " - RTX GPU 최적화"
|
|
echo " - CUDA + TensorRT 가속"
|
|
echo " - 세션 풀: LAMA(3), MIGAN(3), REMBG(2)"
|
|
echo " - 워커: 4-8개 (GPU 메모리에 따라)"
|
|
fi
|
|
|
|
echo ""
|
|
}
|
|
|
|
# 사용자 확인
|
|
confirm_installation() {
|
|
echo "=== 설치 확인 ==="
|
|
echo "다음 설정으로 설치를 진행합니다:"
|
|
echo " 플랫폼: $PLATFORM_NAME"
|
|
echo " 설치 스크립트: $INSTALL_SCRIPT"
|
|
echo ""
|
|
|
|
read -p "계속 진행하시겠습니까? (y/N): " -r
|
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
|
echo "설치가 취소되었습니다."
|
|
exit 0
|
|
fi
|
|
|
|
echo ""
|
|
}
|
|
|
|
# 메인 실행
|
|
main() {
|
|
clear
|
|
print_logo
|
|
|
|
# 스크립트 디렉토리로 이동
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
|
|
cd "$SCRIPT_DIR"
|
|
|
|
detect_platform
|
|
show_gpu_info
|
|
check_requirements
|
|
show_installation_options
|
|
confirm_installation
|
|
|
|
# 플랫폼별 설치 스크립트 실행
|
|
log_step "플랫폼별 설치 스크립트 실행 중..."
|
|
|
|
if [ ! -f "$INSTALL_SCRIPT" ]; then
|
|
log_error "설치 스크립트를 찾을 수 없습니다: $INSTALL_SCRIPT"
|
|
exit 1
|
|
fi
|
|
|
|
chmod +x "$INSTALL_SCRIPT"
|
|
bash "$INSTALL_SCRIPT"
|
|
|
|
# 설치 완료
|
|
echo ""
|
|
echo "=========================================="
|
|
log_success "🎉 인페인팅 서버 설치 완료!"
|
|
echo "=========================================="
|
|
echo ""
|
|
echo "다음 단계:"
|
|
echo " 1. 서버 시작: bash scripts/start_server.sh"
|
|
echo " 2. 서버 상태 확인: bash scripts/status.sh"
|
|
echo " 3. 서버 중지: bash scripts/stop_server.sh"
|
|
echo ""
|
|
echo "접속 주소:"
|
|
echo " 🖥️ 모니터링 대시보드: http://localhost:8009"
|
|
echo " 🔗 API 엔드포인트: http://localhost:8008"
|
|
echo " 📊 헬스체크: curl http://localhost:8008/api/v1/health"
|
|
echo ""
|
|
|
|
if [ "$PLATFORM" = "jetson" ]; then
|
|
echo "Jetson Xavier 참고사항:"
|
|
echo " - tegrastats로 시스템 상태 모니터링 가능"
|
|
echo " - 32GB 통합 메모리로 다중 세션 최적화됨"
|
|
echo " - GPU 온도/클럭 정보가 모니터링 대시보드에 표시됨"
|
|
else
|
|
echo "x86-64 참고사항:"
|
|
echo " - nvidia-smi로 GPU 상태 확인 가능"
|
|
echo " - GPU 메모리에 따라 세션 수 자동 조정됨"
|
|
echo " - TensorRT 사용 시 최고 성능"
|
|
fi
|
|
|
|
echo ""
|
|
}
|
|
|
|
# 스크립트 실행
|
|
main "$@"
|