xichen98cn commited on
Commit
81c8f03
·
verified ·
1 Parent(s): 9489d74

Create cog.yaml

Browse files
Files changed (1) hide show
  1. cog.yaml +28 -0
cog.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ build:
2
+ gpu: true
3
+ cuda: "10.1"
4
+ python_version: "3.8"
5
+ system_packages:
6
+ - "libgl1-mesa-glx"
7
+ - "libglib2.0-0"
8
+ python_packages:
9
+ - "ipython==7.30.1"
10
+ - "numpy==1.21.4"
11
+ - "torch==1.8.1"
12
+ - "torchvision==0.9.1"
13
+ - "opencv-python==4.5.5.62"
14
+ - "Shapely==1.8.0"
15
+ - "h5py==3.6.0"
16
+ - "scipy==1.7.3"
17
+ - "submitit==1.4.1"
18
+ - "scikit-image==0.19.1"
19
+ - "Cython==0.29.27"
20
+ - "timm==0.4.12"
21
+ run:
22
+ - pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html
23
+ - pip install git+https://github.com/cocodataset/panopticapi.git
24
+ - pip install git+https://github.com/mcordts/cityscapesScripts.git
25
+ - git clone https://github.com/facebookresearch/Mask2Former
26
+ - TORCH_CUDA_ARCH_LIST='7.5' FORCE_CUDA=1 python Mask2Former/mask2former/modeling/pixel_decoder/ops/setup.py build install
27
+
28
+ predict: "predict.py:Predictor"