pytorch3d

app.py

@@ -33,12 +33,14 @@ 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 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')
+os.environ["FORCE_CUDA"] = "1"
+os.system('pip install git+https://github.com/facebookresearch/pytorch3d.git@stable')
 
 import numpy as np
 from easydict import EasyDict