diff --git a/docker-compose.yml b/docker-compose.yml index 1a09552..8a1bf4d 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -1,9 +1,15 @@ -version: "3" services: - app: + rag_test: build: . volumes: - .:/opt/workspace # 로컬 디렉토리와 컨테이너 디렉토리 동기화 ports: - - "8501:8501" # Streamlit 앱 포트 설정 - container_name: streamlit-rag # 컨테이너 이름 설정 + - "8502:8501" # Streamlit 앱 포트 설정 + container_name: rag_test # 컨테이너 이름 설정 + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu] \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 171574f..34d2e58 100644 Binary files a/requirements.txt and b/requirements.txt differ