Spaces:
Runtime error
Runtime error
Chao Xu
commited on
Commit
·
d97458e
1
Parent(s):
2ebf2b6
update sparseneus path and set vars
Browse files
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
|