安装maskrcnn-benchmark时遇到的问题

在图片特征提取时需要安装maskrcnn-benchmark包,但是当时遇到了很多问题。下面将总结一下整个安装过程。

根据如下步骤进行安装

(1)git clone https://gitlab.com/vedanuj/vqa-maskrcnn-benchmark.git

(2)cd vqa-maskrcnn-benchmark

(3)git checkout 4c168a637f45dc69efed384c00a7f916f57b25b8 -b stable

(4)python setup.py develop

前面三步安装一切顺利,但是在第四步的时候遇到很多问题:

1、刚开始时我的cuda==10.2,torch==1.7.0+cu102,会出现这个错误:nvcc fatal : Unsupported gpu architecture 'compute_86'

解析一下:就是说我的cuda和torch版本太低了,不支持‘compute_86’,那就进行升级cuda,升级到cuda11.0还是不行,最后一次性升级到cuda11.4。

2、升级cuda11.4时,我根据官网给出的命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

进行安装,这个安装的时候出现错误为:OSError: libmkl_intel_lp64.so: cannot open shared object file: No such file

解析一下:就是pytorch安装不成功。那我就安装cuda11.4对应的pytorch版本torch==1.11.0,但是安装还是出现问题RuntimeError: Error compiling objects for extension,说cuda和torch不匹配。

3、经过一番寻找后,找到了这条命令:

pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f h​​​​​​ttps://download.pytorch.org/whl/cu113/torch_stable.htm​​​​​​lz

通过上述的一番操作之后,cuda和torch版本终于匹配。即完美的解决了cuda和torch不匹配问题。

4、但是在运行 python setup.py develop时又出现了新的问题:error: [Errno 2] No such file or directory: ':/usr/local/cuda-11.4/bin/nvcc'

解析一下:出现的问题是:在运行时cuda路径不对,那就改呗。通过查找发现只需要在你运行程序的环境下运行命令export CUDA_HOME=/usr/local/cuda-11.4即可。

5、解决了所有问题后,运行python setup.py develop,最后出现Finished processing dependencies for maskrcnn-benchmark==0.1,说明安装成功啦。

 

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值