Ubuntu18.04下通过yolov5进行训练(非anaconda版)

本文介绍了在Ubuntu18.04系统上配置Yolov5-6.0环境的步骤,包括安装python3.7、pip,下载并安装模型及依赖。在安装过程中遇到`skbuild`模块缺失的问题,通过更换pip源解决了问题,并成功运行训练命令。作者计划在后续更新中分享如何训练自定义数据集。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近比较忙也没有更新博客,刚好今天给别人配置yolov5的,记录一下吧,希望帮到你们

一、环境

Ubuntu18.04
yolov5-6.0/python3.69

二、前期准备

2.1安装python

 由于Ubuntu18.04自带的python版本是3.6.9对应的yolov5是6.0版本的,如果想要运行高版本的则需要下载3.6以上版本的python

sudo apt-get install python3.7 #根据yolov5版本选择python版本,我的python版本是3.6.9,所以我直接下载的yolov5-6.0
python --version

2.2安装pip(选择python3的)

sudo apt install python-pip3  #(python3)
sudo apt install python pip   #(python2)

2.3下载yolov5模型

 这里我直接把u盘里的拷进来了,不会的可以去查一下怎么下载的,这里就不赘述了

2.3.1 安装包

进入到yolov5文件夹中

cd yolov5
pip3 install -U -r requirements.txt  -i https://pypi.tuna.tsinghua.edu.cn/simple

但是在安装包的过程中我出了错

ImportError: No module named skbuild

解决方法:换源最后发现豆瓣的源是可以的

pip install scikit-build -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

2.4 训练

进入yolo文件夹中,打开终端

输入指令

python3 train.py --img 640 --batch 2 --epoch 3 --data coco128.yaml --weights yolov5s.pt

最后,就可以跑通啦~

下篇更新如何训练自己的数据集!

### 安装和运行 YOLOv8 框架 YOLOv8 是 Ultralytics 提供的一个最新本的目标检测框架,其安装过程相对简单,但仍需一些必要的准备工作。以下是关于如何在 Ubuntu 18.04安装和运行 YOLOv8 的详细说明。 #### 环境准备 为了成功部署 YOLOv8,在开始之前需要确认以下几点: - **操作系统**: 已经安装Ubuntu 18.04。 - **硬件需求**: 建议使用 NVIDIA GPU 来加速模型训练和推理[^3]。 - **软件依赖**: - Python (推荐本 >= 3.8) - pip 和其他开发工具链(如 GCC) #### 步骤一:更新系统并安装必要工具 确保系统的包管理器是最新的,并安装一些常用的开发工具: ```bash sudo apt update && sudo apt upgrade -y sudo apt install build-essential cmake git wget unzip python3-pip -y ``` 上述命令会安装 `build-essential`、`cmake`、`git` 等基础工具以及 Python 的 pip 包管理器[^2]。 #### 步骤二:安装 Anaconda 或 Miniconda 虽然可以直接使用系统自带的 Python,但建议通过 Conda 创建独立的虚拟环境来隔离同项目的依赖关系。可以按照以下步骤安装 Miniconda: ```bash wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh bash Miniconda3-latest-Linux-x86_64.sh source ~/.bashrc ``` 完成后可以通过以下命令验证安装是否成功: ```bash conda --version ``` 如果显示本号,则表示安装成功。 #### 步骤三:创建并激活 Conda 虚拟环境 创建一个新的 Conda 环境,并指定所需的 Python 本(Python>=3.8): ```bash conda create -n yolov8_env python=3.9 conda activate yolov8_env ``` #### 步骤四:安装 PyTorch 及 CUDA 支持 根据你的 GPU 驱动本选择合适的 PyTorch 和 CUDA 组合。例如,假设你已经安装了 CUDA 11.7,那么可以通过以下命令安装 PyTorch: ```bash pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 ``` 如果没有 GPU 或者想使用 GPU 加速,可以选择 CPU-only 本: ```bash pip install torch torchvision torchaudio ``` 更多选项可以在官方页面找到:[PyTorch Installation](https://pytorch.org/get-started/locally/) [^3]。 #### 步骤五:克隆 YOLOv8 仓库并安装依赖 从 GitHub 克隆 YOLOv8 的代码库,并进入该目录: ```bash git clone https://github.com/ultralytics/ultralytics.git cd ultralytics/ ``` 接着安装所需的所有 Python 库: ```bash pip install -r requirements.txt ``` 这一步会自动安装诸如 NumPy、Pillow、Matplotlib 等常用库[^4]。 #### 步骤六:验证安装 完成以上所有步骤后,可以通过简单的脚本来测试 YOLOv8 是否正常工作。例如,尝试加载预训练权重并对图片进行目标检测: ```python from ultralytics import YOLO # Load a model model = YOLO('yolov8n.pt') # load an official model # Use the model for inference results = model('https://ultralytics.com/images/bus.jpg') ``` 保存这段代码到文件中并通过 Python 执行它。如果一切顺利,你应该能看到一张带有标注框的结果图[^1]。 --- ### 注意事项 - 如果计划使用 GPU 进行训练或推断,请务必先检查显卡驱动程序是否已正确安装,并通过 `nvidia-smi` 查看当前状态。 - 对于大规模数据集或者复杂模型架构来说,可能还需要额外调整内存分配策略或其他超参数设置。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

科研分母

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值