ubuntu18 YOLOv5 配置及训练

最新推荐文章于 2025-11-05 08:15:10 发布

原创

最新推荐文章于 2025-11-05 08:15:10 发布 · 6.8k 阅读

105 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #pytorch

本文详细介绍了如何在Ubuntu 18上安装Anaconda，配置环境，创建独立环境，安装PyTorch和YOLOv5，并解决了训练过程中的常见错误。步骤包括下载脚本、环境设置、创建环境、安装依赖、测试模型和数据集训练。

ubuntu18 YOLOv5 配置及训练

环境搭建

#下载anaconda安装脚本（本教程采用清华源）
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh
#安装anaconda
bash Anaconda3-2020.07-Linux-x86_64.sh
    
# Please, press ENTER to continue -> 回车继续
# 阅读协议，同意按回车（跳过ctrl+c）
# Do you accept the license terms? [yes|no] -> 同意协议输入yes回车
# Anaconda3 will now be installed into this location -> 选择安装路径（本文安装/usr/local/anaconda3），等待安装
# by running conda init? [yes|no] -> 是否添加系统环境，输入yes回车
# 待安装结束

# 刷新当前用户环境（激活环境）
source ~/.bashrc

#更新Anaconda相关配置
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

创建环境

#使用conda创建单独的环境，在系统命令行中运行如下命令：
conda create -n yolov5_1 python==3.8
#进入该环境继续安装依赖库：
conda activate yolov5_1
#删除环境
conda remove -n yolov5 --all
#退出当前环境
conda deactivate yolov5_1


#安装pytorch环境，可能需要等待
#pytorch的gpu版本，官方网站地址: https://pytorch.org/get-started/locally/，找到自己匹配的相关命令
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.</