|
在调试RandLA-Net pytorch版本的代码时,按照要求安装了torch-points-kernels库,在运行model.py时,在导入库的语句提示错误:
import torch
import torch.nn as nn
try:
from torch_points 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文件夹拷贝到自己的项目下,即可解决问题。
|