update

app.py

@@ -4,9 +4,21 @@ import gradio as gr
 from PIL import Image
 from functools import partial
 import argparse
+import sys
+import torch
+
+if os.getenv('SYSTEM') == 'spaces':
+ os.system('pip install --global-option="--no-networks" git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
 
-os.system('pip install --global-option="--no-networks" git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch')
-os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"')
+ pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
+ version_str="".join([
+ f"py3{sys.version_info.minor}_cu",
+ torch.version.cuda.replace(".",""),
+ f"_pyt{pyt_version_str}"
+ ])
+ os.system('pip install fvcore iopath')
+ os.system(f'pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html')
+ # os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"')
 
 import cv2
 import time
@@ -16,7 +28,6 @@ from segment_anything import sam_model_registry, SamPredictor
 
 import random
 from pytorch3d import transforms
-import torch
 import torchvision
 import torch.distributed as dist
 import nvdiffrast.torch as dr