Chao Xu commited on
Commit
d492a58
โ€ข
1 Parent(s): b6f95eb

torchsparse debug

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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"], env={"FORCE_CUDA": "1"})
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