mike23415 commited on
Commit
6d2bc0f
·
verified ·
1 Parent(s): 804b2ab

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -10
Dockerfile CHANGED
@@ -1,23 +1,22 @@
1
- FROM python:3.10-slim
2
 
3
  WORKDIR /app
4
 
5
- # Install system packages
6
  RUN apt-get update && apt-get install -y \
7
  libgl1 \
8
  libglib2.0-0 \
9
  && rm -rf /var/lib/apt/lists/*
10
 
11
- # Copy requirements early
12
- COPY requirements.txt ./
13
-
14
- # Step 1: install numpy first to avoid torch import error
15
- RUN pip install --no-cache-dir numpy
16
 
17
- # Step 2: install torch and torchvision (CPU versions)
18
- RUN pip install --no-cache-dir torch==2.0.1+cpu torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
 
19
 
20
- # Step 3: install everything else
 
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
23
  # Copy application code
 
1
+ FROM python:3.10
2
 
3
  WORKDIR /app
4
 
5
+ # System dependencies
6
  RUN apt-get update && apt-get install -y \
7
  libgl1 \
8
  libglib2.0-0 \
9
  && rm -rf /var/lib/apt/lists/*
10
 
11
+ # Ensure pip is up to date
12
+ RUN pip install --upgrade pip
 
 
 
13
 
14
+ # Install numpy first, then torch/torchvision (CPU)
15
+ RUN pip install numpy
16
+ RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
17
 
18
+ # Copy requirements and install the rest
19
+ COPY requirements.txt ./
20
  RUN pip install --no-cache-dir -r requirements.txt
21
 
22
  # Copy application code