marcosremar2 commited on
Commit
2dbea89
·
1 Parent(s): 76390d9

Update PDF to Markdown converter API with NVIDIA L4 support

Browse files
Files changed (1) hide show
  1. 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.1+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 && \
 
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 && \