About PVCNN

[201722111112027]

Hi, I encountered some problems when reproducing the code. The model stopped when loading into pvcnn/modules/functional/backend.py. I used python version 3.9, torch version 2.4.0, cuda version 12.1, gcc-12. As following:

import os
from pathlib import Path
from torch.utils.cpp_extension import load

gcc_path = os.getenv('CC', default='/usr/bin/gcc')
if not Path(gcc_path).is_file():
raise ValueError('Could not find your gcc, please replace it here.')

_src_path = os.path.dirname(os.path.abspath(file))
_backend = load(
name='_pvcnn_backend',
extra_cflags=['-O3', '-std=c++17'],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-12'],
sources=[os.path.join(_src_path,'src', f) for f in [
'ball_query/ball_query.cpp',
'ball_query/ball_query.cu',
'grouping/grouping.cpp',
'grouping/grouping.cu',
'interpolate/neighbor_interpolate.cpp',
'interpolate/neighbor_interpolate.cu',
'interpolate/trilinear_devox.cpp',
'interpolate/trilinear_devox.cu',
'sampling/sampling.cpp',
'sampling/sampling.cu',
'voxelization/vox.cpp',
'voxelization/vox.cu',
'bindings.cpp',
]]
)

all = ['_backend']