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

文章描述了在Ubuntu20.04上安装Detectron2时遇到的CUDA版本不匹配问题,以及如何通过检查和修改环境变量,确保CUDA版本正确,并成功安装Detectron2的步骤。关键在于调整环境变量至匹配pytorch所依赖的CUDA版本,并通过`pip`安装Detectron2。
摘要由CSDN通过智能技术生成

在安装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,请参考本站其他内容

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值