Update app.py

app.py

@@ -37,8 +37,11 @@ opts.save_dir = './'
 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')
-# spaces.GPU(os.system("FORCE_CUDA=1 pip install 'git+https://github.com/facebookresearch/pytorch3d.git'"))
+# os.system('pip install iopath') FORCE_CUDA=1 
+os.system('curl -LO https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz')
+os.system('tar xzf 1.10.0.tar.gz')
+os.system('export CUB_HOME=$PWD/cub-1.10.0')
+spaces.GPU(os.system("pip install 'git+https://github.com/facebookresearch/pytorch3d.git'"))
 os.system("mkdir -p checkpoints/ && wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/")
 
 print(f'>>> System info: CUDA: {torch.version.cuda}')