[人工智能]cuda10.1 + py3.7 + torch1.7.1 安装MinkowskiEngine

新建conda环境

conda create -n mink python=3.7
source activate mink

安装torch1.7.1+torchvision0.8.2，可到此处下载

torch-1.7.1+cu101-cp37-cp37m-linux_x86_64.whl
torchvision-0.8.2+cu101-cp37-cp37m-linux_x86_64.whl

在whl路径下安装

pip install torch-1.7.1+cu101-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.8.2+cu101-cp37-cp37m-linux_x86_64.whl

安装后注意torch的cuda版本是否为10.1，可通过下面命令查看

python -c "import torch; print(torch.version.cuda)"

安装依赖库

sudo apt-get install libopenblas-dev

安装open3d

pip install open3d

最后，安装MinkowskiEngine

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