ubuntu20.04安装MinkowskiEngine

ubuntu20.04安装MinkowskiEngine

安装这个库之前首先确认环境中pytorch的cudatoolkit版本是否与安装的CUDA版本一致，确保一致，然后开始安装

我的CUDA版本是11.6,故采用下面的安装命令先安装pytroch及对应工具包

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

安装好了之后测试一下，确保可以调用GPU

#测试一下，确保cuda可以使用
python
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()

安装依赖

pip install ninja  #官方文档没说，这里依赖还需要安装ninja库
conda install openblas-devel -c anaconda  #安装依赖

我在安装openblas-devel时，它自动又安装了cpu版本的pytorch，从而导致pytorch无法调用GPU，torch.cuda.is_available()为False，我删掉环境重新安装pytorch和这个依赖都是如此，无论是先安装openblas-devel依赖还是先安装GPU版本的pytorch，都出现pytorch安装成cpu版本的问题。后面解决办法是清楚conda中下载的包的缓存，让这个依赖真正重新下载而不是从cache中安装，发现这次不会附加安装cpu版本的pytorch了，很奇怪。。。

然后将仓库clone下来本地安装

git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine
python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas

报错，没有找到CUDA的位置

因为我之前安装了CUDA11.6,所以需要通过export命令指定CUDA的位置（这里根据自己安装CUDA的位置来改）

export CUDA_HOME=/usr/local/cuda-11.6

然后因为这个issue，需要将MAX_JOBS设置为2再进行安装

export MAX_JOBS=2

然后执行安装命令

python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas

安装成功

安装完成之后测试一下是否能正常导入MinkowskiEngine

python
>>> import MinkowskiEngine as ME
>>> print(ME.__version__)
0.5.4

收工