YiftachEde commited on
Commit
017eabf
·
1 Parent(s): 3951932
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ cuda_version_str = torch.version.cuda.replace(".", "") if torch.cuda.is_availabl
17
  version_str = f"py3{sys.version_info.minor}_cu{cuda_version_str}_pyt{pyt_version_str}"
18
 
19
  print(f"Looking for PyTorch3D wheel for: {version_str}")
20
- os.system(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
21
 
22
  # If the wheel installation fails, try installing from source
23
  try:
 
17
  version_str = f"py3{sys.version_info.minor}_cu{cuda_version_str}_pyt{pyt_version_str}"
18
 
19
  print(f"Looking for PyTorch3D wheel for: {version_str}")
20
+ os.system(f"pip install \"git+https://github.com/facebookresearch/pytorch3d.git\"")
21
 
22
  # If the wheel installation fails, try installing from source
23
  try: