89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
환경 설정 확인 스크립트
|
|
"""
|
|
|
|
def check_installation():
|
|
print("🔍 설치된 패키지 확인 중...")
|
|
|
|
try:
|
|
import torch
|
|
print(f"✅ PyTorch: {torch.__version__}")
|
|
|
|
# Apple Silicon MPS 확인
|
|
if torch.backends.mps.is_available():
|
|
print("✅ Apple Silicon MPS 가속 사용 가능")
|
|
else:
|
|
print("⚠️ MPS 가속 사용 불가 (CPU 모드)")
|
|
|
|
except ImportError:
|
|
print("❌ PyTorch 설치 실패")
|
|
return False
|
|
|
|
try:
|
|
import transformers
|
|
print(f"✅ Transformers: {transformers.__version__}")
|
|
except ImportError:
|
|
print("❌ Transformers 설치 실패")
|
|
return False
|
|
|
|
try:
|
|
import sentencepiece
|
|
print("✅ SentencePiece 설치 완료")
|
|
except ImportError:
|
|
print("❌ SentencePiece 설치 실패")
|
|
return False
|
|
|
|
try:
|
|
import accelerate
|
|
print("✅ Accelerate 설치 완료")
|
|
except ImportError:
|
|
print("❌ Accelerate 설치 실패")
|
|
return False
|
|
|
|
# 문서 처리 라이브러리 확인
|
|
doc_libs = []
|
|
try:
|
|
import PyPDF2
|
|
doc_libs.append("PyPDF2")
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
import pdfplumber
|
|
doc_libs.append("pdfplumber")
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
from docx import Document
|
|
doc_libs.append("python-docx")
|
|
except ImportError:
|
|
pass
|
|
|
|
if doc_libs:
|
|
print(f"✅ 문서 처리: {', '.join(doc_libs)}")
|
|
else:
|
|
print("⚠️ 문서 처리 라이브러리 설치 필요")
|
|
|
|
# 시스템 정보
|
|
print(f"\n📊 시스템 정보:")
|
|
print(f" Python 버전: {torch.version.python}")
|
|
if torch.backends.mps.is_available():
|
|
print(f" 디바이스: Apple Silicon (MPS)")
|
|
else:
|
|
print(f" 디바이스: CPU")
|
|
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
print("🚀 NLLB 번역 시스템 환경 확인")
|
|
print("=" * 40)
|
|
|
|
if check_installation():
|
|
print("\n🎉 환경 설정 완료!")
|
|
print("다음 단계: NLLB 모델 다운로드")
|
|
else:
|
|
print("\n❌ 환경 설정 실패")
|
|
print("패키지 재설치 필요")
|