在此列出在我们CentOS 7.8.2003系统中使用的包:
- CUDA 10.1 (cudatoolkit 10.1.243)
- python 3.7.10
- GCC 7.5
- pytorch 1.6.0
- torchvision 0.7.0
- numpy 1.20.2
- mkl-include 2021.2.0
- minkowskiengine 0.4.3
- plyfile 0.7.4
- torch_scatter 2.0.7
- open3d 0.9.0
对我们的系统安装这些包的步骤如下:
conda create -n py3-mink-43 python=3.7
conda activate py3-mink-43
conda install numpy==1.20.2
conda install mkl-include==2021.2.0
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
- 从这里下载0.4.3版本的Minkowski Engine的
.zip
安装压缩包并解压,然后使用如下命令安装。
cd MinkowskiEngine-0.4.3
python setup.py install --force_cuda --cuda_home=<cuda path>
- 安装一些额外需要的包,命令如下。
pip install plyfile==0.7.4
pip install open3d==0.9.0
请确保你使用了0.9.0版本的open3d,否则你会在运行程序时遇到问题。