一、系统环境
Ubuntu 18.04 + cuda 10.2 + python 3.6
二、Yolo V5
单独安装numpy、torch和torchision。
numpy: pip3 install numpy
torch、torchision: 参考博客:https://blog.csdn.net/comli_cn/article/details/107468083
(我安装了torch1.7.0 + torchvision0.8.1)
注释掉requirement.txt中的numpy、torch和torchvisio:
pip3 install -r requirements.txt
遇到的问题以及解决办法:
1、import 报错 Illegal instruction(core dumped)
原因:numpy版本冲突
解决方法:
'''修改环境变量'''
sudo gedit /etc/profile
把 export OPENBLAS_CORETYPE=ARMV8 加入最后面一行,然后保存
'''更新环境变量'''
source /etc/profile
2、安装torchision报错:fatal error: libavcodec/avcodec.h: No such file or directory
解决方法:参考https://shliang.blog.csdn.net/article/details/109862917
三、Paddle
下载对应版本的Paddle whl文件:https://www.paddlepaddle.org.cn/documentation/docs/zh/2.0/install/Tables.html
TX2 是arm架构 ,找到对应cuda版本:
参考教程:https://aistudio.baidu.com/aistudio/projectdetail/1591846