Spaces:
Runtime error
Runtime error
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
|