alrichardbollans commited on
Commit
0f2eb17
·
1 Parent(s): 89ed204

Try docker for detectron

Browse files
Files changed (2) hide show
  1. Dockerfile +7 -1
  2. requirements.txt +6 -1
Dockerfile CHANGED
@@ -1,4 +1,6 @@
1
- FROM python:3.12
 
 
2
 
3
  WORKDIR /code
4
 
@@ -6,6 +8,10 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
 
 
 
 
9
  COPY . .
10
 
11
  EXPOSE 7860
 
1
+ FROM docker.io/library/python:3.10@sha256:0a886c151ca7a5deffce7ec5c8b519e75cf238274232119d9d791dc22b0d0fd3
2
+
3
+ RUN apt-get update && apt-get install -y git git-lfs ffmpeg libsm6 libxext6 cmake rsync libgl1-mesa-glx && rm -rf /var/lib/apt/lists/* && git lfs install
4
 
5
  WORKDIR /code
6
 
 
8
 
9
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
10
 
11
+ RUN pip install --no-cache-dir pip -U && pip install --no-cache-dir datasets "huggingface-hub>=0.19" "hf_xet>=1.0.0,<2.0.0" "hf-transfer>=0.1.4" "protobuf<4" "click<8.1" "pydantic~=1.0"
12
+
13
+ RUN pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git
14
+
15
  COPY . .
16
 
17
  EXPOSE 7860
requirements.txt CHANGED
@@ -3,4 +3,9 @@ shiny
3
  shinywidgets
4
  plotly
5
  pandas
6
- ridgeplot
 
 
 
 
 
 
3
  shinywidgets
4
  plotly
5
  pandas
6
+ ridgeplot
7
+ opencv-python-headless
8
+ pyyaml==5.1
9
+ torch
10
+ torchvision
11
+ ninja