Ubuntu20安装onnx遇到的几个问题及解决方法

30 篇文章 3 订阅
21 篇文章 1 订阅
在Ubuntu上使用Docker构建目标检测算法环境时,通过国内镜像源安装requirements.txt时遇到onnx安装问题。首先解决cmake缺失,然后因找不到onnx 1.8.1版本,安装protobuf库后成功安装onnx。
摘要由CSDN通过智能技术生成

在Ubuntu上用Docker容器安装一个目标检测算法的运行环境,过程中出了几个小问题,记录一下。

工程的requirements.txt如下:

numpy
torch>=1.7
opencv_python
loguru
scikit-image
tqdm
torchvision
Pillow
thop
ninja
tabulate
tensorboard

# verified versions
# pycocotools corresponds to https://github.com/ppwwyyxx/cocoapi
pycocotools>=2.0.2
onnx==1.8.1
onnxruntime==1.8.0
onnx-simplifier==0.3.5

使用国内镜像源安装:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

安装到onnx时出现如下错误:

两个错误提示,一个是找不到cmake,一个是找不到onnx 1.8.1版本。先解决cmake的问题,安装cmake:

conda install cmake

 安装完成后,再次执行安装onnx的命令:

pip install onnx==1.8.1

这次不报版本问题了,但是报了一个PEP 517的错误: 

百度搜了一下说是onnx依赖protobuf,所以需要先安装protobuf的库:

apt-get install libprotobuf-dev protobuf-compiler

安装完成后,再次安装onnx,成功了。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值