torch_points_kernels遭遇 ModuleNotFoundError: No module named ‘torch_points_kernels.points_cpu‘

在调试RandLA-Net pytorch版本的代码时,按照要求安装了torch-points-kernels库,在运行model.py时,在导入库的语句提示错误:

import torch
import torch.nn as nn

try:
    from torch_points import knn
    #from torch_points_kernels import knn
except (ModuleNotFoundError, ImportError):
    from torch_points_kernels import knn

错误提示:

ModuleNotFoundError: No module named 'torch_points_kernels.points_cpu'

反复安装了各个版本的torch-points-kernels库,均没有解决问题,之后通过手动编译torch-points-kernels库解决问题,方法如下:
1.进入项目页面克隆项目:https://github.com/torch-points3d/torch-points-kernels
2.安装库:

pip install torch-points-kernels

3.在本地编译项目:

python setup.py build_ext --inplace
python -m unittest

(上述步骤在项目README.md中都有说明)
4.将编译得到的torch_points_kernels文件夹拷贝到自己的项目下,即可解决问题。

你可能感兴趣的:(深度学习,计算机视觉,人工智能,神经网络,pytorch)