francischung222 commited on
Commit
aee5df5
·
1 Parent(s): 34337a4

2nd fix buggy install in requirements.txt

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -1
Dockerfile CHANGED
@@ -5,12 +5,16 @@ WORKDIR /app
5
  # Install system dependencies
6
  RUN apt-get update && apt-get install -y \
7
  git \
 
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
  # Copy requirements and install Python packages
11
  COPY requirements.txt .
12
  # Install torch from CPU wheels explicitly to avoid source builds
13
- RUN pip install --no-cache-dir torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu
 
 
 
14
  RUN pip install --no-cache-dir -r requirements.txt
15
 
16
  # Copy application code
 
5
  # Install system dependencies
6
  RUN apt-get update && apt-get install -y \
7
  git \
8
+ libgomp1 \
9
  && rm -rf /var/lib/apt/lists/*
10
 
11
  # Copy requirements and install Python packages
12
  COPY requirements.txt .
13
  # Install torch from CPU wheels explicitly to avoid source builds
14
+ RUN pip install --no-cache-dir \
15
+ torch==2.2.2 \
16
+ --index-url https://download.pytorch.org/whl/cpu \
17
+ --extra-index-url https://pypi.org/simple
18
  RUN pip install --no-cache-dir -r requirements.txt
19
 
20
  # Copy application code