Spaces:
Runtime error
Runtime error
Chao Xu
commited on
Commit
·
2aac31f
1
Parent(s):
febb747
env variable
Browse files- env.sh +4 -0
- requirements.txt +2 -1
env.sh
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
|
| 3 |
+
export TORCH_CUDA_ARCH_LIST="{archs}"
|
| 4 |
+
export IABN_FORCE_CUDA=1
|
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
|
| 2 |
# sparseneus
|
|
|
|
| 3 |
-e git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0#egg=torchsparse
|
| 4 |
opencv_python
|
| 5 |
trimesh
|
|
@@ -9,5 +9,6 @@ icecream
|
|
| 9 |
tqdm
|
| 10 |
scipy
|
| 11 |
PyMCubes
|
|
|
|
| 12 |
# sudo apt-get install libsparsehash-dev
|
| 13 |
inplace_abn
|
|
|
|
|
|
|
| 1 |
# sparseneus
|
| 2 |
+
-r env.sh
|
| 3 |
-e git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0#egg=torchsparse
|
| 4 |
opencv_python
|
| 5 |
trimesh
|
|
|
|
| 9 |
tqdm
|
| 10 |
scipy
|
| 11 |
PyMCubes
|
| 12 |
+
ninja
|
| 13 |
# sudo apt-get install libsparsehash-dev
|
| 14 |
inplace_abn
|