Huiwenshi commited on
Commit
099bf18
·
verified ·
1 Parent(s): 1e1ed99

Upload app_hg.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app_hg.py +1 -1
app_hg.py CHANGED
@@ -68,7 +68,7 @@ def install_cuda_toolkit():
68
  # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
69
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
70
 
71
- subprocess.call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"], env=env)
72
  import pytorch3d
73
 
74
  install_cuda_toolkit()
 
68
  # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
69
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
70
 
71
+ subprocess.call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
72
  import pytorch3d
73
 
74
  install_cuda_toolkit()