Noursine commited on
Commit
f2726bd
·
verified ·
1 Parent(s): b37ad86

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -10,9 +10,11 @@ RUN apt-get update && apt-get install -y \
10
  libsm6 \
11
  libxext6 \
12
  g++ \
 
 
 
13
  && rm -rf /var/lib/apt/lists/*
14
 
15
-
16
  # Upgrade pip
17
  RUN pip install --upgrade pip
18
 
@@ -22,9 +24,6 @@ RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu --index-url https://dow
22
  # Install Python dependencies
23
  RUN pip install -r requirements.txt
24
 
25
- # Create a writable gdown cache directory
26
- RUN mkdir -p /.cache/gdown && chmod -R 777 /.cache
27
-
28
  # Install Detectron2 from GitHub
29
  RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
30
 
 
10
  libsm6 \
11
  libxext6 \
12
  g++ \
13
+ build-essential \
14
+ cmake \
15
+ python3-dev \
16
  && rm -rf /var/lib/apt/lists/*
17
 
 
18
  # Upgrade pip
19
  RUN pip install --upgrade pip
20
 
 
24
  # Install Python dependencies
25
  RUN pip install -r requirements.txt
26
 
 
 
 
27
  # Install Detectron2 from GitHub
28
  RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
29