torchcpu

Author: YH0223

Dockerfile

@@ -11,7 +11,10 @@ RUN apt-get update && apt-get install -y \
 # pyproject.toml 복사 및 의존성 설치
 COPY pyproject.toml .
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir .
+    pip install --no-cache-dir torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
+
+# 3. 나머지 종속성 설치 (이때 pyproject.toml의 torch는 이미 설치되어 패스됨)
+RUN pip install --no-cache-dir .
 
 # 애플리케이션 코드 복사
 COPY . .