pointMLP-pytorch icon indicating copy to clipboard operation
pointMLP-pytorch copied to clipboard

Unable to import pointnet2_utils

Open vae-forever opened this issue 5 months ago • 1 comments

Thanks for sharing. However I'm having trouble configuring the environment: Unable to import pointnet2_utils in pointnet2_modules.py

vae-forever avatar Jan 23 '24 09:01 vae-forever

Thanks for sharing. first of all, I used colab to run this code.

%cd /content/pointMLP-pytorch-main/pointnet2_ops_lib !python setup.py install --verbose

Before using the above command to install you must change the setup.py file from lines 19-36 with the below code:

os.environ["TORCH_CUDA_ARCH_LIST"] = "7.5" setup( name="pointnet2_ops", version=__version__, author="Erik Wijmans", packages=find_packages(), install_requires=requirements, ext_modules=[ CUDAExtension( name="pointnet2_ops._ext", sources=_ext_sources, extra_compile_args={ "cxx": ["-O3"], "nvcc": ["-O3", "-Xfatbin", "-compress-all", "-v"], }, include_dirs=[osp.join(this_dir, _ext_src_root, "include")], ) ]

Also, before running the main code, you must modify the pointnet2_utils.py file from lines 23 to 31 with the provided code below.

os.environ["TORCH_CUDA_ARCH_LIST"] = "7.5" _ext = load( "_ext", sources=_ext_sources, extra_include_paths=[osp.join(_ext_src_root, "include")], extra_cflags=["-O3"], extra_cuda_cflags=["-O3", "-Xfatbin", "-compress-all", "-v"], with_cuda=True, )

I hope you will successfully execute the code.

rezamoradi avatar Mar 14 '24 02:03 rezamoradi