ubuntu配置yolov5环境

本文硬件平台为工控机，系统环境为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版本进行安装
1
2
3
4
5

安装成功了会提示Successfully installed XXX
可以输入python进入pyrhon环境，导包测试，没有报错和能输出版本信息则安装成功了。

3. 安装opencv

sudo apt-get install python3-opencv
1

如果出现找不到命令，先更新一下sudo apt-get update

4.下载yolov5源码

直接去github上下载就可以了。

5.安装yolov5环境

• 打开下载的源码目录里面的requirements.txt文件，注释掉opencv-python，torch，torchvison这三行，因为已经单独安装了
• 输入命令安装其余包

sudo pip install -U -r requirements.txt 
1

6.安装完成

• 安装完成后运行如果报错未找到XXX包，直接再用pip安装对应包即可