update

app.py

@@ -6,7 +6,6 @@ import spaces
 # os.system('pip install iopath')
 # os.system("pip install -v -v -v  'git+https://github.com/facebookresearch/pytorch3d.git@stable'")
 # os.system("cd pytorch3d && pip install -e . && cd ..")
-os.system("mkdir -p checkpoints/ && wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/")
 
 import gradio as gr
 import random
@@ -39,7 +38,9 @@ os.makedirs(opts.save_dir,exist_ok=True)
 test_tensor = torch.Tensor([0]).cuda()
 opts.device = str(test_tensor.device)
 os.system('pip install iopath')
-os.system("FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='6.0;6.1;7.0;7.5;8.0;8.6+PTX' CUDA_HOME='/usr/local/cuda' pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'")
+os.system("FORCE_CUDA=1 pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'")
+os.system("mkdir -p checkpoints/ && wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/")
+
 from viewcrafter import ViewCrafter
 
 def viewcrafter_demo(opts):