haotiz commited on
Commit
95c19fe
·
1 Parent(s): 708dec4

add cuda specific requirement

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. requirement.txt +21 -0
app.py CHANGED
@@ -10,9 +10,9 @@ import warnings
10
 
11
  warnings.filterwarnings("ignore")
12
 
13
- os.system(
14
- "pip install einops shapely timm yacs tensorboardX ftfy prettytable pymongo click opencv-python inflect nltk scipy scikit-learn pycocotools")
15
- os.system("pip install transformers")
16
  os.system("python setup.py build develop --user")
17
 
18
  from maskrcnn_benchmark.config import cfg
 
10
 
11
  warnings.filterwarnings("ignore")
12
 
13
+ # os.system(
14
+ # "pip install einops shapely timm yacs tensorboardX ftfy prettytable pymongo click opencv-python inflect nltk scipy scikit-learn pycocotools")
15
+ # os.system("pip install transformers")
16
  os.system("python setup.py build develop --user")
17
 
18
  from maskrcnn_benchmark.config import cfg
requirement.txt ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch
3
+ torchvision
4
+ torchaudio
5
+ einops
6
+ shapely
7
+ timm
8
+ yacs
9
+ tensorboardX
10
+ ftfy
11
+ prettytable
12
+ pymongo
13
+ click
14
+ opencv-python
15
+ inflect
16
+ nltk
17
+ scipy
18
+ scikit-learn
19
+ pycocotools
20
+ transformers
21
+ git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI&egg=pycocotools