Spaces:
Runtime error
Runtime error
Chao Xu
commited on
Commit
Β·
b5a2293
1
Parent(s):
d74847a
change torchsparse installation
Browse files- app.py +2 -0
- requirements.txt +1 -1
app.py
CHANGED
@@ -23,6 +23,8 @@ if not is_local_run:
|
|
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 |
|
27 |
import inspect
|
28 |
import shutil
|
|
|
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(["FORCE_CUDA=1", "pip", "install", "--no-cache-dir", "git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0"])
|
28 |
|
29 |
import inspect
|
30 |
import shutil
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# sparseneus
|
2 |
-
-e git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0#egg=torchsparse
|
3 |
opencv_python
|
4 |
trimesh
|
5 |
numpy
|
|
|
1 |
# sparseneus
|
2 |
+
# -e git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0#egg=torchsparse
|
3 |
opencv_python
|
4 |
trimesh
|
5 |
numpy
|