Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -12,8 +12,8 @@ RUN apt-get update && apt-get install -y \
|
|
| 12 |
libgomp1 \
|
| 13 |
&& rm -rf /var/lib/apt/lists/*
|
| 14 |
|
| 15 |
-
# 2. Install PyTorch (CPU)
|
| 16 |
-
RUN pip install --no-cache-dir torch torchvision torchaudio --index-url
|
| 17 |
|
| 18 |
# 3. Install Requirements
|
| 19 |
COPY requirements.txt .
|
|
|
|
| 12 |
libgomp1 \
|
| 13 |
&& rm -rf /var/lib/apt/lists/*
|
| 14 |
|
| 15 |
+
# 2. Install PyTorch (CPU)
|
| 16 |
+
RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 17 |
|
| 18 |
# 3. Install Requirements
|
| 19 |
COPY requirements.txt .
|