PVD/metrics/ChamferDistancePytorch/chamfer3D/setup.py

16 lines
459 B
Python
Raw Normal View History

2021-10-19 20:54:46 +00:00
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
name='chamfer_3D',
ext_modules=[
CUDAExtension('chamfer_3D', [
"/".join(__file__.split('/')[:-1] + ['chamfer_cuda.cpp']),
"/".join(__file__.split('/')[:-1] + ['chamfer3D.cu']),
]),
],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'],
cmdclass={
'build_ext': BuildExtension
})