ubuntu中通过源码安装pointnet2_ops_lib

注：本帖所用环境为：ubuntu 24.04、 cuda 12.04

文章目录

• 1. 克隆 `PointNet++` 源码库
• 2. 安装依赖
• 3. 编译 `pointnet2_ops_lib`
• 4. 测试安装

1. 克隆 PointNet++ 源码库

首先，克隆 PointNet++ 的 GitHub 仓库：

git clone https://github.com/erikwijmans/Pointnet2_PyTorch.git
cd Pointnet2_PyTorch

2. 安装依赖

确保你已经安装了 PyTorch，并且系统已经正确配置了 CUDA（如果你打算使用 GPU 加速）。你可以通过以下命令来安装必要的依赖：

pip install -r requirements.txt

这个安装过程可能出现如下问题：

这是由于当前cuda版本比较新导致的，这个时候可以修改pointnet2_ops_lib/setup.py中找到类似以下部分的代码：

extra_compile_args={
    'cxx': ['-O2'],
    'nvcc': [
        '-O2',
        '-arch=compute_37',
        ...
    ]
}

你需要将 compute_37 改为 CUDA 12.4 支持的架构代码，例如 compute_80 或更高：

'nvcc': ['-arch=compute_80', '-arch=sm_80', ...]

常见的 CUDA 架构如下：

• compute_50: Maxwell 架构
• compute_52: Maxwell 架构
• compute_60: Pascal 架构
• compute_61: Pascal 架构，例如Tesla P40
• compute_70: Volta 架构
• compute_75: Turing 架构
• compute_80: Ampere 架构，例如RTX4090

3. 编译 pointnet2_ops_lib

pip install -e .
pip install pointnet2_ops_lib/.

4. 测试安装

编译和安装完成后，可以测试库是否安装成功。运行以下 Python 代码以确保 pointnet2_ops_lib 可正常导入：

import pointnet2_ops_lib

如果没有报错，说明安装成功。