Chao Xu commited on
Commit
b6f95eb
·
1 Parent(s): b5a2293

correct torchsparse install

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ if not is_local_run:
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
 
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(["pip", "install", "--no-cache-dir", "git+https://github.com/mit-han-lab/[email protected]"], env={"FORCE_CUDA": "1"})
28
 
29
  import inspect
30
  import shutil