Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y \
|
|
| 12 |
COPY requirements.txt .
|
| 13 |
|
| 14 |
# Install PyTorch with CPU-specific index
|
| 15 |
-
RUN pip install --no-cache-dir torch==2.4.0
|
| 16 |
|
| 17 |
# Install remaining dependencies
|
| 18 |
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
| 12 |
COPY requirements.txt .
|
| 13 |
|
| 14 |
# Install PyTorch with CPU-specific index
|
| 15 |
+
RUN pip install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
|
| 16 |
|
| 17 |
# Install remaining dependencies
|
| 18 |
RUN pip install --no-cache-dir -r requirements.txt
|