Sunnycodes3114 commited on
Commit
369e582
·
verified ·
1 Parent(s): 598dda7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # Base image for Python
2
  FROM python:3.10-slim
3
 
4
  # Install system dependencies
@@ -17,7 +17,7 @@ RUN apt-get update && apt-get install -y \
17
  # Upgrade pip
18
  RUN pip install --no-cache-dir --upgrade pip
19
 
20
- # Install PyTorch and torchvision first
21
  RUN pip install --no-cache-dir torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
22
 
23
  # Install Detectron2
@@ -27,7 +27,7 @@ RUN python -m pip install 'git+https://github.com/facebookresearch/detectron2.gi
27
  COPY requirements.txt /tmp/requirements.txt
28
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
29
 
30
- # Set the working directory
31
  WORKDIR /app
32
 
33
  # Copy application files
 
1
+ # Base Python image
2
  FROM python:3.10-slim
3
 
4
  # Install system dependencies
 
17
  # Upgrade pip
18
  RUN pip install --no-cache-dir --upgrade pip
19
 
20
+ # Install PyTorch and torchvision
21
  RUN pip install --no-cache-dir torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
22
 
23
  # Install Detectron2
 
27
  COPY requirements.txt /tmp/requirements.txt
28
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
29
 
30
+ # Set working directory
31
  WORKDIR /app
32
 
33
  # Copy application files