ubuntu配置yolov5环境

本文硬件平台为工控机,系统环境为ubuntu18

配置yolov5步骤

1.下载pytorch和torchvision软件包
  • 由于在线安装容易出现安装失败,所以本文使用的是本地安装。本文是基于miniconda安装的,miniconda安装参考之前的博客:ubuntu中安装miconda。

  • pytorch和torchvision下载链接:potorch和torchvision下载链接

ubuntu配置yolov5环境_第1张图片

  • 进去分别下载torch和torchvision,然后根据自己python版本和硬件系统和架构下载对应wel包。本工控机是X86和ubuntu,conda下安装的是python3.8,所以下载的版本如下:
    在这里插入图片描述
    在这里插入图片描述
  • python、pytorch、torchvision版本对应关系:

ubuntu配置yolov5环境_第2张图片

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环境,导包测试,没有报错和能输出版本信息则安装成功了。

ubuntu配置yolov5环境_第3张图片

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安装对应包即可

你可能感兴趣的:(ubuntu,ubuntu,linux,YOLO)