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
|