update

app.py

@@ -24,6 +24,7 @@ def install_cuda_toolkit():
 install_cuda_toolkit()
 
 os.system('pip install -e ./thirdparty/DROID-SLAM')
+os.system('pip install -e ./thirdparty/DROID-SLAM/thirdparty/lietorch')
 import torch
 import numpy as np
 from easydict import EasyDict

thirdparty/DROID-SLAM/setup.py

@@ -30,32 +30,32 @@ setup(
     cmdclass={ 'build_ext' : BuildExtension }
 )
 
-setup(
-    name='lietorch',
-    version='0.2',
-    description='Lie Groups for PyTorch',
-    packages=['lietorch'],
-    package_dir={'': 'thirdparty/lietorch'},
-    ext_modules=[
-        CUDAExtension('lietorch_backends', 
-            include_dirs=[
-                osp.join(ROOT, 'thirdparty/lietorch/lietorch/include'), 
-                osp.join(ROOT, 'thirdparty/eigen')],
-            sources=[
-                'thirdparty/lietorch/lietorch/src/lietorch.cpp', 
-                'thirdparty/lietorch/lietorch/src/lietorch_gpu.cu',
-                'thirdparty/lietorch/lietorch/src/lietorch_cpu.cpp'],
-            extra_compile_args={
-                'cxx': ['-O2'], 
-                'nvcc': ['-O2',
-                    '-gencode=arch=compute_60,code=sm_60', 
-                    '-gencode=arch=compute_61,code=sm_61', 
-                    '-gencode=arch=compute_70,code=sm_70', 
-                    '-gencode=arch=compute_75,code=sm_75',
-                    '-gencode=arch=compute_80,code=sm_80',
-                    '-gencode=arch=compute_86,code=sm_86',                 
-                ]
-            }),
-    ],
-    cmdclass={ 'build_ext' : BuildExtension }
-)
+# setup(
+#     name='lietorch',
+#     version='0.2',
+#     description='Lie Groups for PyTorch',
+#     packages=['lietorch'],
+#     package_dir={'': 'thirdparty/lietorch'},
+#     ext_modules=[
+#         CUDAExtension('lietorch_backends', 
+#             include_dirs=[
+#                 osp.join(ROOT, 'thirdparty/lietorch/lietorch/include'), 
+#                 osp.join(ROOT, 'thirdparty/eigen')],
+#             sources=[
+#                 'thirdparty/lietorch/lietorch/src/lietorch.cpp', 
+#                 'thirdparty/lietorch/lietorch/src/lietorch_gpu.cu',
+#                 'thirdparty/lietorch/lietorch/src/lietorch_cpu.cpp'],
+#             extra_compile_args={
+#                 'cxx': ['-O2'], 
+#                 'nvcc': ['-O2',
+#                     '-gencode=arch=compute_60,code=sm_60', 
+#                     '-gencode=arch=compute_61,code=sm_61', 
+#                     '-gencode=arch=compute_70,code=sm_70', 
+#                     '-gencode=arch=compute_75,code=sm_75',
+#                     '-gencode=arch=compute_80,code=sm_80',
+#                     '-gencode=arch=compute_86,code=sm_86',                 
+#                 ]
+#             }),
+#     ],
+#     cmdclass={ 'build_ext' : BuildExtension }
+# )