Spaces:
Paused
Paused
Chao Xu
commited on
Commit
·
d492a58
1
Parent(s):
b6f95eb
torchsparse debug
Browse files
app.py
CHANGED
@@ -22,9 +22,10 @@ if not is_local_run:
|
|
22 |
# export IABN_FORCE_CUDA=1
|
23 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
24 |
os.environ["IABN_FORCE_CUDA"] = "1"
|
|
|
25 |
subprocess.run(["pip", "install", "inplace_abn"])
|
26 |
# FORCE_CUDA=1 pip install --no-cache-dir git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0
|
27 |
-
subprocess.run(["pip", "install", "--no-cache-dir", "git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0"]
|
28 |
|
29 |
import inspect
|
30 |
import shutil
|
|
|
22 |
# export IABN_FORCE_CUDA=1
|
23 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
24 |
os.environ["IABN_FORCE_CUDA"] = "1"
|
25 |
+
os.environ["FORCE_CUDA"] = "1"
|
26 |
subprocess.run(["pip", "install", "inplace_abn"])
|
27 |
# FORCE_CUDA=1 pip install --no-cache-dir git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0
|
28 |
+
subprocess.run(["pip", "install", "--no-cache-dir", "git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0"])
|
29 |
|
30 |
import inspect
|
31 |
import shutil
|