Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
@@ -31,7 +31,10 @@ WORKDIR /app
|
|
31 |
# Copy requirements first for better caching
|
32 |
COPY requirements.txt .
|
33 |
|
34 |
-
# Install
|
|
|
|
|
|
|
35 |
RUN pip3 install --no-cache-dir -r requirements.txt
|
36 |
|
37 |
# Copy application code
|
|
|
31 |
# Copy requirements first for better caching
|
32 |
COPY requirements.txt .
|
33 |
|
34 |
+
# Install PyTorch CPU version first
|
35 |
+
RUN pip3 install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
|
36 |
+
|
37 |
+
# Install other Python dependencies
|
38 |
RUN pip3 install --no-cache-dir -r requirements.txt
|
39 |
|
40 |
# Copy application code
|