一、起因
最近在复现Kpconv和Regtr时需要用到MinkowskiEngine库,上一次安装就是在网上找了好多教程,勉强安好,这次又是安了两天,所以记录一下。
二、过程记录
1 准备工作
①先确定cuda版本:
nvcc -V
我的是
建议cuda版本不要太高
②检查gcc/g++
gcc --version
g++ --version
③虚拟环境是必须的,我列出我的以供参考,也可以根据源码的来。
第二次的配置:
torch 1.11.0 ;torchaudio 0.11.0;torchvision 0.12.0;Python 3.8.18
第一次的配置:
torch 1.13.1+cu116
torchaudio 0.13.1+cu116
torchvision 0.14.1+cu116
Python 3.9.19
2 安装pytorch3d
选择本地安装
pytorch3d下载地址:Release Version 0.6.2 · facebookresearch/pytorch3d · GitHub
下载这个
上传服务器解压安装即可
3 安装ninja
普通pip安装即可,有的博客讲需要安装openblas-devel,我没有安装成功这个依赖,但也成功安装ME了。
4 安装MinknowskiEngine
还是选择本地安装
MinkowskiEngine地址:GitHub - NVIDIA/MinkowskiEngine: Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
进入Mink文件夹下输入
python setup.py install --blas_include_dirs=${CONDA_X}/include --blas=openblas
安装成功
三、我的报错
ERROR: Could not find a version that satisfies the requirement MinknowskiEngine (from versions: none)
ERROR: No matching distribution found for MinknowskiEngine
Retrieving notices: ...working... done
Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
PackagesNotFoundError: The following packages are not available from current channels:
- minknowskiengine
还有个报错找不到了 后面能找到就再补上。
四、参考
官网指南:
一些博客:
Ubuntu22.04安装MinkowskiEngine0.5.4_ubuntu安装minkowskiengine-CSDN博客
REGTR环境配置(MinkowskiEngine安装) - 知乎 (zhihu.com)