cuhgrel commited on
Commit
5ed11a5
·
1 Parent(s): 94ad879

updated Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -34,11 +34,11 @@
34
  # Upgrade pip first
35
  RUN pip install --no-cache-dir --upgrade pip setuptools wheel
36
 
37
- # Install PyTorch with its specific index-url FIRST
38
  RUN pip install --no-cache-dir \
39
- torch==2.0.1 \
40
- torchaudio==2.0.2 \
41
- --index-url https://download.pytorch.org/whl/cpu
42
 
43
  # Install all other requirements from the file in a single step
44
  # Use --use-deprecated=legacy-resolver if dependency conflicts are complex
 
34
  # Upgrade pip first
35
  RUN pip install --no-cache-dir --upgrade pip setuptools wheel
36
 
37
+ # Install PyTorch with its specific index-url INSTALL FIRST BEFORE REQUIREMENTS.TXT
38
  RUN pip install --no-cache-dir \
39
+ --index-url https://download.pytorch.org/whl/cpu \
40
+ torch==2.1.0 \
41
+ torchaudio==2.1.0
42
 
43
  # Install all other requirements from the file in a single step
44
  # Use --use-deprecated=legacy-resolver if dependency conflicts are complex