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
收工