pytorch3d

app.py

@@ -34,8 +34,9 @@ def install_cuda_toolkit():
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 
-print("Compling DROID-SLAM")
+print("Compling other packages")
 install_cuda_toolkit()
+os.system('pip install git+https://github.com/facebookresearch/pytorch3d.git@stable')
 os.system('pip install ./thirdparty/DROID-SLAM')
 os.system('pip install ./thirdparty/DROID-SLAM/thirdparty/lietorch')
 

requirements.txt

@@ -1,3 +1,2 @@
 chumpy@git+https://github.com/mattloper/chumpy
 torch-scatter==2.1.2
-git+https://github.com/facebookresearch/pytorch3d.git@stable