RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine

RTX3090 + cuda 11.1 + torch1.9.0 安装 MinkowskiEngine

转载自 https://blog.csdn.net/weixin_43486780/article/details/119849000

创建conda环境

conda create -n mink python=3.8
source activate mink

  
  

  
  

安装torch1.9.0 + torchvision

可到此处下载whl文件（也可以直接pip安装）

pip install torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl

  
  

  
  

然后安装torchvision

pip install torchvision

  
  

  
  

安装完后要看torch对应的cuda版本是否为cuda11.1

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

  
  

  
  

安装open3d

pip install open3d

  
  

  
  

安装MinkowskiEngine

下载MinkowskiEngine代码库并安装

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

  
  

  
  

可能出现的错误

fatal error: cblas.h: No such file or directory

解决办法

sudo apt-get install libopenblas-dev