Update app.py

app.py

@@ -38,16 +38,22 @@ 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') 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('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'"))
+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}"
+])
+print(version_str)
+os.system(f"{sys.executable} -m pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
 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}')
-os.system('ls -l /usr/local | grep cuda')
-os.system('nvcc -V')
-os.system('nvidia-smi')
+print(f'>>> System info: {version_str}')
+
 
 from viewcrafter import ViewCrafter