Spaces:
Paused
Paused
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/[email protected]
|
| 27 |
+
subprocess.run(["FORCE_CUDA=1", "pip", "install", "--no-cache-dir", "git+https://github.com/mit-han-lab/[email protected]"])
|
| 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/[email protected]#egg=torchsparse
|
| 3 |
opencv_python
|
| 4 |
trimesh
|
| 5 |
numpy
|
|
|
|
| 1 |
# sparseneus
|
| 2 |
+
# -e git+https://github.com/mit-han-lab/[email protected]#egg=torchsparse
|
| 3 |
opencv_python
|
| 4 |
trimesh
|
| 5 |
numpy
|