개발 환경 구성

프로젝트 구성

프로젝트 디렉토리를 만들고, 다음과 같이 4개 파일을 생성합니다(통크게 디렉토리명을 pytorch-vision-api로 지어봤습니다.🤣):

pytorch-vision-api/
├── app.py       # FastAPI 메인 파일
├── model.py     # 모델 로드 및 추론 함수
├── utils.py     # 유틸리티 함수
└── requirements.txt # 의존성 패키지 목록

필요 패키지 설치

먼저 requirements.txt 파일의 내용을 아래와 같이 작성하고, 이를 사용하여 설치해보겠습니다:

# PyTorch 및 torchvision
torch==2.5.0
torchvision==0.20.0

# FastAPI
fastapi==0.115.5
uvicorn==0.32.1
python-multipart==0.0.17

파일을 저장한 다음, 필요 패키지를 설치합니다:

pip install -r requirements.txt

별다른 에러 없이 설치가 완료된다면 다음으로 넘어가보겠습니다.