本文硬件平台为工控机,系统环境为ubuntu18
配置yolov5步骤
1.下载pytorch和torchvision软件包
-
由于在线安装容易出现安装失败,所以本文使用的是本地安装。本文是基于miniconda安装的,miniconda安装参考之前的博客:ubuntu中安装miconda。
-
pytorch和torchvision下载链接:potorch和torchvision下载链接
- 进去分别下载torch和torchvision,然后根据自己python版本和硬件系统和架构下载对应wel包。本工控机是X86和ubuntu,conda下安装的是python3.8,所以下载的版本如下:
- python、pytorch、torchvision版本对应关系:
2.ubuntu中安装torch和torchvision
安装命令
pip install torch-1.7.1+cpu-cp38-cp38-linux_x86_64
pip install torchvision-0.8.1+cpu-cp38-cp38-linux_x86_64
# 这里我在安装torchvision过程中发现在重新下载torch 1.7.0版本
# 猜想是版本没有对应上
# 所以我终止了下载,重新下载的torchvision 0.8.2版本进行安装
安装成功了会提示Successfully installed XXX
可以输入python进入pyrhon环境,导包测试,没有报错和能输出版本信息则安装成功了。
3. 安装opencv
sudo apt-get install python3-opencv
4.下载yolov5源码
直接去github上下载就可以了。
5.安装yolov5环境
- 打开下载的源码目录里面的requirements.txt文件,注释掉opencv-python,torch,torchvison这三行,因为已经单独安装了
- 输入命令安装其余包
sudo pip install -U -r requirements.txt
6.安装完成
- 安装完成后运行如果报错未找到XXX包,直接再用pip安装对应包即可