在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,成功了。