Satyajithchary commited on
Commit
3f4a2a2
1 Parent(s): 4feeaf1

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +18 -0
Dockerfile ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Install dependencies
2
+ RUN apt-get update && apt-get install -y \
3
+ python3-dev \
4
+ build-essential \
5
+ libjpeg-dev \
6
+ zlib1g-dev \
7
+ libpng-dev \
8
+ libglib2.0-0 \
9
+ libsm6 \
10
+ libxrender1 \
11
+ libxext6 \
12
+ && rm -rf /var/lib/apt/lists/*
13
+
14
+ # Install PyTorch and TorchVision
15
+ RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
16
+
17
+ # Install Detectron2 from source
18
+ RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'