MinkowskiEngine安装

MinkowskiEngine(ME)是计算稀疏张量常用的库,但是官方文档里安装这个库的坑还是蛮多的。

参考了好多经验贴,终于在自己电脑上成功安装编译MinkowskiEngine,uu们务必看完本贴再跟着操作,有些问题放在后面解释了

在这里插入图片描述

参考:
https://blog.csdn.net/m0_60197472/article/details/125293739

环境准备

我的环境:

ubuntu18.04
cuda版本:cuda 11.3
gcc版本:7.5.0
g++版本:7.5.0

linux查询指令:

nvcc -V
gcc --version
g++ --version

查询结果:
在这里插入图片描述

在这里插入图片描述

anaconda安装

官方教程

Anaconda
MinkowskiEngine supports both CUDA 10.2 and cuda 11.1, which work for most of latest pytorch versions.
我根据此教程流程安装:
在这里插入图片描述
1、这部分顺利安装
当然根据上面的经验贴,更改了安装顺序,先安装pytorch,然后再安装依赖

conda create -n 虚拟环境名 python=3.7.0
conda activate 虚拟环境名

从现在开始,所有指令走在虚拟环境中执行


# 去官网找对应的安装指令,我使用的指令与官方教程不同
conda install pytorch=1.8.0 torchvision cudatoolkit=11.1 -c pytorch -c nvidia
conda install openblas-devel -c anaconda

安装好pytorch后,最好测试一下
我一开始安装成了cpu版本的导致出错,这里可以去官网找对应的安装指令(我使用pytorch 1.8.0,cudatoolkit 11.1)
在这里插入图片描述

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

2、开始出问题了

接下来到了Install MinkowskiEngine环节,网络问题选择本地安装
将代码下载后打开MinkowskiEngine目录

# Or if you want local MinkowskiEngine
# 这里我手动下载解压的
git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine

在执行这行python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas命令时,出现了两个普遍的问题。

问题1:subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1.

看着两位大神的解释:
在这里插入图片描述
在这里插入图片描述
解决方案:
1、 降低版本是必不可能的
2、 修改torch中的将 [‘ninja’,‘-v’]改成[‘ninja’,‘–version’] (确实解决了这个报错,但是导致程序根本不编译,后续会找不到 .o 文件,网上也很难找到别人编译好的 .o 文件,因此此方法无效)
(PS:这里还需要安装ninja才会走到这步)pip install ninja # 官方文档没说,这里依赖还需要安装ninja库
在这里插入图片描述
3、将MinkowskiEngine目录中 setup.py 文件中的“cmdclass={‘build_ext’: BuildExtension}”这一行改为“cmdclass={‘build_ext’: BuildExtension.with_options(use_ninja=False)}”,pytorch默认使用ninjia作为backend,这里把它禁用掉就好了;(之前的ninja白装了)
在这里插入图片描述
第一个报错解决后,又出来新问题:

问题2:error: command ‘:/usr/local/cuda/bin/nvcc’ failed: No such file or directory: ‘:/usr/local/cuda/bin/nvcc’

在这里插入图片描述
在对应路径下能找到nvcc,判断应该是环境变量路径设置有误
解决方案:
在这里插入图片描述
这里还有一个坑,修改环境变量后,回到了base环境,需要重新进入自己搭建的虚拟环境中运行 setup.py。

细节

执行 setup.py 需要一段时间
同时,如果在自己电脑上运行有可能会卡住
编译过程如果卡死,在setup文件里面142行左右,把12改小(我改成了4);
在这里插入图片描述

编译过程会报一堆warning,不影响安装,无所谓的;
安装成功后测试一下:

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

在这里插入图片描述

  • 20
    点赞
  • 41
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值