openvino2022.1.0+yolov5-6.1,ubuntu18.04下运行环境搭建

openvino2022.1.0+yolov5-6.1,ubuntu18.04下运行环境搭建:

1.安装OpenVINO Runtime运行库

pip3 install openvino==2022.1.0

测试环境,如果安装成功则控制台不会有任何输出

python3 -c "from openvino.runtime import Core"

如何找不到安装包,则可能需要更新pip

python3 -m pip3 install --upgrade pip

2.安装核显驱动

from openvino.runtime import Core

ie = Core()
devices = ie.available_devices

for device in devices:
    device_name = ie.get_property(device_name=device, name="FULL_DEVICE_NAME")
    print(f"{device}: {device_name}")

使用上述代码测试可用设备发现只有cpu可以,此时需要更新显卡驱动
在这里插入图片描述
将驱动安装文件,放置在目录下
在这里插入图片描述
neo目录中包含手动下载好的驱动文件
在这里插入图片描述
设置执行权限

chmod 755 install_NEO_OCL_driver.sh

运行安装脚本(此脚本在官方脚本的基础上进行了修改,注释了下载驱动包的部分)

sudo -E ./install_NEO_OCL_driver.sh

安装完成后再次测试可用设备,发现GPU可用。
在这里插入图片描述

在这里插入图片描述
3.安装依赖

pip3 install pyyaml

升级opencv版本

pip3 install opencv-python==4.5.5.64

扩展阅读

在i7-6700TCPU的工控机运行,对比C++版本,和python版本,调用openvino加速包对同一张图片循环加载并推理的运行情况

代码推理设备cpu占用平均耗时
c++cpu350%216ms
c++gpu102%212ms
pythoncpu361%208ms
pythongpu100%220ms
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值