Chao Xu commited on
Commit
d97458e
·
1 Parent(s): 2ebf2b6

update sparseneus path and set vars

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -12,8 +12,15 @@ sys.path.append(code_dir)
12
  elev_est_dir = os.path.join(code_dir, "one2345_elev_est/")
13
  sys.path.append(elev_est_dir)
14
 
 
 
 
15
  import subprocess
16
  subprocess.run(["sh", os.path.join(elev_est_dir, "install.sh")], cwd=elev_est_dir)
 
 
 
 
17
  subprocess.run(["pip", "install", "inplace_abn"])
18
 
19
  import inspect
 
12
  elev_est_dir = os.path.join(code_dir, "one2345_elev_est/")
13
  sys.path.append(elev_est_dir)
14
 
15
+ sparseneus_dir = os.path.join(code_dir, "SparseNeuS_demo_v1/")
16
+ sys.path.append(sparseneus_dir)
17
+
18
  import subprocess
19
  subprocess.run(["sh", os.path.join(elev_est_dir, "install.sh")], cwd=elev_est_dir)
20
+ # export TORCH_CUDA_ARCH_LIST="7.0;7.2;8.0;8.6"
21
+ # export IABN_FORCE_CUDA=1
22
+ os.environ["TORCH_CUDA_ARCH_LIST"] = "7.0;7.2;8.0;8.6"
23
+ os.environ["IABN_FORCE_CUDA"] = "1"
24
  subprocess.run(["pip", "install", "inplace_abn"])
25
 
26
  import inspect