Spaces:
Sleeping
Sleeping
Commit
·
2dbea89
1
Parent(s):
76390d9
Update PDF to Markdown converter API with NVIDIA L4 support
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -33,7 +33,7 @@ COPY requirements.txt .
|
|
33 |
|
34 |
# Install PyTorch dependencies first
|
35 |
RUN pip install --no-cache-dir --upgrade pip && \
|
36 |
-
pip install --no-cache-dir torch==2.5.
|
37 |
|
38 |
# Install remaining Python dependencies
|
39 |
RUN grep -v "torch\|torchvision\|torchaudio" requirements.txt > requirements_no_torch.txt && \
|
|
|
33 |
|
34 |
# Install PyTorch dependencies first
|
35 |
RUN pip install --no-cache-dir --upgrade pip && \
|
36 |
+
pip install --no-cache-dir torch==2.5.0+cu118 torchvision==0.20.0+cu118 torchaudio==2.5.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
|
37 |
|
38 |
# Install remaining Python dependencies
|
39 |
RUN grep -v "torch\|torchvision\|torchaudio" requirements.txt > requirements_no_torch.txt && \
|