这两天在复现点云相关的一些论文,发现好多项目都是要用到pointnet这个基础的模型,而且许多都是把pointnet中的FPS集成到了cuda上,所以需要进行一系列配置。
根据这个项目进行配置。https://github.com/erikwijmans/Pointnet2_PyTorch
但是在输入pip install -r requirements安装 pointnet2_ops_lib/.时,出现报错ERROR: Failed building wheel for pointnet2-ops。
查阅了github中的相关讨论区,发现是torch版本的问题。
正确的解决步骤如下:
1、conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7 -c pytorch -c nvidia
安装上述版本的torch。
2、
export LD_LIBRARY_PATH="/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda-11.7/bin:$PATH"
导入以上两个环境变量。
可以成功安装pointnet2-ops。
原文连接如下: