Omkar008 commited on
Commit
7d5f1b5
·
verified ·
1 Parent(s): f9479b4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -29,7 +29,7 @@ COPY . /app
29
  # Install Python dependencies
30
  RUN pip install --no-cache-dir -U pip setuptools wheel
31
  RUN pip install --no-cache-dir setuptools-rust
32
- RUN pip install --no-cache-dir torch==2.0.1+cpu torchaudio==2.0.2+cpu -f https://download.pytorch.org/whl/cpu.html
33
  RUN pip install --no-cache-dir -r requirements.txt
34
 
35
  # Create and switch to a non-root user
 
29
  # Install Python dependencies
30
  RUN pip install --no-cache-dir -U pip setuptools wheel
31
  RUN pip install --no-cache-dir setuptools-rust
32
+ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
33
  RUN pip install --no-cache-dir -r requirements.txt
34
 
35
  # Create and switch to a non-root user