Detectron2安装报错问题(本地cuda)

在安装Detectron2遇到了困难,如此报错:

 Could not build wheels for detection2,which is required to install pyproject.toml-based projects

或者是:

detectron2 RuntimeError: Error compiling objects for extension

或者是:

This is an issue with the package mentioned above, not pip.

排除版本问题:

首先去官方说明查看detection2要求:

Installation — detectron2 0.6 documentation

         简而言之:

        ①使用macOS或是Linux操作系统,且python版本≥3.7

        ②pytorch版本≥1.8,且安装对应版本的torchvision以及cuda

        ③若需要可安装opencv

        ④gcc与g++版本≥5.4,安装可以看别人的教程,注意更改默认版本

【仅供参考】本人系统Ubuntu20.04,其它版本:

        ①python = 3.7.0

        ②torch = 1.12.0,torchvision = 0.13.0,cuda = 10.2

        ③gcc&g++ = 5.4

问题解决:

①检查.bashrc

sudo gedit ~/.bashrc

②找到环境变量

可以看到我调用的本地cuda有问题,我安装的pytorch是cuda10.2版本的,那我就需要把路径改为正常路径(cuda-10.2),修改上述四个含有cuda-11.6的内容

③修改完保存关闭,并使更新立即生效

source ~/.bashrc

④查看当前使用的cuda版本

nvcc --version

如果显示为更换后的版本,即代表成功

⑤执行detectron安装命令

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'

⑥successfully,问题解决

另外,本文只针对本地cuda,如若使用pytorch自带cuda,请参考本站其他内容

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值