yucornetto commited on
Commit
1c48e2c
1 Parent(s): fafc670

init for demo

Browse files

Change-Id: I9de14e43b257432e36218b5dba6ceaacccd4b789

Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ os.system("python -m pip install 'git+https://github.com/facebookresearch/detect
9
 
10
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
11
 
 
 
12
  import gradio as gr
13
  # check pytorch installation:
14
  from detectron2.utils.logger import setup_logger
 
9
 
10
  os.system("pip install git+https://github.com/cocodataset/panopticapi.git")
11
 
12
+ os.system("TORCH_CUDA_ARCH_LIST='7.5' FORCE_CUDA=1 python fcclip/modeling/pixel_decoder/ops/setup.py build install")
13
+
14
  import gradio as gr
15
  # check pytorch installation:
16
  from detectron2.utils.logger import setup_logger