快速入门

安装

可以通过 pip 或 conda 安装 MinkowskiEngine。 目前，安装要求如下：

  • Ubuntu 14.04 或更高版本

  • CUDA 10.1 或更高版本（如果您需要 CUDA 加速）

  • pytorch 1.3 或更高版本

  • python 3.6 或更高版本

  • GCC 6 或更高版本

系统要求

MinkowskiEngine 需要 openblas、python3-dev 以及 torch、numpy python 包。 强烈建议使用 anaconda，以下说明将安装所有要求。

安装

MinkowskiEngine 通过 PyPI MinkowskiEngine 分发，可以使用 pip 简单地安装。

pip3 install -U MinkowskiEngine

要安装最新版本，请使用 pip3 install -U git+https://github.com/NVIDIA/MinkowskiEngine。

运行分割网络

下载 MinkowskiEngine 并运行示例代码。

git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine
python -m examples.indoor

当您运行以上示例时，它将下载 Minkowski 网络的预训练权重，并可视化 3D 场景的语义分割结果。

仅 CPU 编译

git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine
python setup.py install --cpu_only

其他 BLAS 和 MKL 支持

在 intel CPU 设备上，conda 安装的 numpy 带有 Intel Math Kernel Library 或 MKL。 Minkowski Engine 将使用 numpy 自动检测 MKL，并使用 MKL 而不是 openblas 或 atlas。

在许多情况下，这将自动完成。 但是，如果 numpy 未使用 MKL，但您拥有 Intel CPU，请使用 conda 安装 MKL。

conda install -c intel mkl mkl-include
python setup.py install --blas=mkl

如果要在 MKL、ATLAS、OpenBLAS 和系统 BLAS 中使用特定的 BLAS，请按如下所示提供 blas 名称

cd MinkowskiEngine
python setup.py install --blas=openblas