Chao Xu commited on
Commit
b5a2293
β€’
1 Parent(s): d74847a

change torchsparse installation

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. 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