HuguesTHOMAS / KPConv-PyTorch

Kernel Point Convolution implemented in PyTorch
MIT License
786 stars 157 forks source link

update ? #249

Closed 945fc41467 closed 4 months ago

945fc41467 commented 4 months ago

Hello,

Currently, the program doesn’t compile with python>3.10 and numpy>1.26 (and I am not sure about cuda12).

Would it be possible to update it ?

945fc41467 commented 4 months ago

After some tries, I was able to use it with cuda 12.

For the compilation under python3.12, I had to modify the two setups.py files :

# cpp_subsamplin/setup.py:
from setuptools import setup, Extension
import numpy as np

# Adding OpenCV to project
# ************************

# Adding sources of the project
# *****************************

SOURCES = ["../cpp_utils/cloud/cloud.cpp",
             "grid_subsampling/grid_subsampling.cpp",
             "wrapper.cpp"]

module = Extension(name="grid_subsampling",
                    sources=SOURCES,
                    extra_compile_args=['-std=c++11',
                                        '-D_GLIBCXX_USE_CXX11_ABI=0'])

setup(ext_modules=[module], include_dirs=np.get_include())

and

# cpp_neighbors/setup.py:
from setuptools import setup, Extension
import numpy as np

# Adding OpenCV to project
# ************************

# Adding sources of the project
# *****************************

SOURCES = ["../cpp_utils/cloud/cloud.cpp",
             "neighbors/neighbors.cpp",
             "wrapper.cpp"]

module = Extension(name="radius_neighbors",
                    sources=SOURCES,
                    extra_compile_args=['-std=c++11',
                                        '-D_GLIBCXX_USE_CXX11_ABI=0'])

setup(ext_modules=[module], include_dirs=np.get_include())

didn’t try to launch a training with this yet. But there’s no reason.