Huiwenshi commited on
Commit
a7aa7d6
1 Parent(s): 5e9f7a7

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
  try:
71
- subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
72
  import pytorch3d
73
  except Exception as err:
74
  print(err)
 
68
  # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
69
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
70
  try:
71
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "git+https://github.com/facebookresearch/pytorch3d@stable"])
72
  import pytorch3d
73
  except Exception as err:
74
  print(err)