目录
1. 查看电脑状况
深度学习的训练对于电脑显卡要求较高,若电脑没有独立显卡(NVIDIA卡)或者是AMD的显卡无法使用GPU进行训练,需要用CPU进行训练,会慢很多。若不清楚电脑有无显卡可以打开任务管理器,点击性能往下找GPU,就可以看到是否有独立显卡,不清楚是否为独立显卡则搜索型号。
若有独立显卡则安装GPU版本的pytorch,若无独立显卡则安装CPU版本的pytorch。
点击下载训练源码 夸克网盘下载 ,建议先全部转存提前下载,若有需要下载的资源失效,可至公众号获取百度盘链接下载。
YOLOv8网络结构图,论文必备,无水印图可 微信公众号-笑脸惹桃花 回复“888” 获取。
2. 安装所需软件
推荐安装Anaconda3+Pycharm,都需要加入环境变量,会安装或者安装过了则跳过这一步骤。
2.1 Anaconda3安装
Anaconda3由于是国外网站下载较慢,推荐通过清华镜像源安装。
Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
选择带有Anaconda3...Windows...exe 字样的进行下载,或者点击网盘下载相同版本。
下载完成之后打开Anaconda3进行安装,一直点下一步,选Just Me,安装路径不建议安装到c盘,可以直接复制粘贴修改到 D:\Anaconda3 ,也可以修改到其他路径,最好纯英文路径。
点击下一步后,需要选择添加到环境变量,如下图前三个一定要勾选,也可以按照我图中全选。
点击install安装后耐心等待进度条满安装完成即可,进度较慢,耐心等待。
2.2 Pycharm安装
安装Pycharm可以直接去官网下载,速度较快。
下载 PyCharm:JetBrains 出品的用于数据科学和 Web 开发的 Python IDE
往下拉下载第二个Community Edition社区免费版就可以,或者点击网盘下载相同版本。
下载完之后打开安装,点下一步,遇到选择路径 修改路径到D盘或者其它除C盘外的文件夹,可以建个自己喜欢的英文名字。
需要勾选这些选项,五角星必勾选,建议全选。
再点下一步,直接安装就可以了,耐心等待进度条满安装完成即可。
3. 安装环境
3.1 安装cuda及cudnn
3.1.1 下载及安装cuda
在安装pytorch前需要安装cuda,(若无显卡或intel的显卡则跳过,直接看4.4使用CPU训练)下载cuda前需要先查看显卡支持的CUDA版本最高是多少,按下win+r键,输入cmd,在打开的页面输入:nvidia-smi ,即可查看。
上图红框位置显示即为cuda最高支持版本,本教程cuda最高版本达到11.8即可,若没有达到则需要更新显卡驱动。
CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive
下载完之后运行安装,建议安装到默认路径,所以C盘需要留有20G以上的存储空间,一直点击下一步,直到出现这个界面点击自定义,然后全部勾选即可。
将下图中选项全部勾选安装。
安装完成后可以再次在cmd里输入命令:nvcc -V 查看,如下显示即安装成功
3.1.2 cudnn安装
进入cudnn官网,选择合适版本的文件。
https://developer.nvidia.com/rdp/cudnn-archivehttps://developer.nvidia.com/rdp/cudnn-archive
进入后在文件列表中选择cudnn版本与上面cuda安装相匹配的版本。
下载Windows版本的压缩包文件。
下载需要登录,也可以复制下载链接打开迅雷下载,或者点击链接下载下载cudnn。
将得到的压缩文件进行解压,解压后得到下图三个文件夹,全选复制进cuda的文件夹中进行覆盖替换,替换完成后即cudnn安装完成。按照本文教程安装的cuda的文件夹默认在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8 目录下。
3.2 创建虚拟环境
按下Win键,输入anaconda prompt,打开下图所示的应用。
默认进入的是base环境,base环境的版本与下载的anaconda3版本有关,因此不建议直接使用,需要新建环境,在新建环境之前建议更改默认的pip源和conda源可加速下载速度。
更改代码如下,直接输入即可,这里选择更换的是中科大源,建议使用,亲测完美运行。
conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple
之后通过pip install 包名 方式安装的任何库(除了pytorch及相关库),只要报错ERROR: No matching distribution found for 包名 ,在python版本与本文一致的情况下,皆为源的访问速度慢导致的,可以通过以下方法临时换源下载安装,方法为 在安装包时通过 -i
参数指定镜像源,可多试几个确定哪个速度最快最稳定
pip install 包名 -i https://pypi.tuna.tsinghua.edu.cn/simple
清华大学
URL:https://pypi.tuna.tsinghua.edu.cn/simple
阿里云
URL:https://mirrors.aliyun.com/pypi/simple
中国科学技术大学(USTC)
URL:https://pypi.mirrors.ustc.edu.cn/simple
豆瓣(Douban)
URL:https://pypi.doubanio.com/simple
华为云
URL:https://repo.huaweicloud.com/repository/pypi/simple
腾讯云
URL:https://mirrors.cloud.tencent.com/pypi/simple
如:
pip install numpy==1.26.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
此时新建虚拟环境(需要关闭加速软件),这里创建一个名为yolov8,python版本为3.9的虚拟环境,也可以修改为其他名,本文所用为python3.9,同时本教程所用pytorch源及python版本只支持3.9。
conda create -n yolov8 python=3.9
回车后出现新建环境提醒输入y继续,耐心等待全部下载完成后自动安装。
注意:此时如果报错
UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel anaconda/pkgs/free <https://mirrors.ustc.edu.cn/anaconda/pkgs/free>
The channel is not accessible or is invalid.
You will need to adjust your conda configuration to proceed.
Use `conda config --show channels` to view your configuration's current state,
and use `conda config --show-sources` to view config file locations.
等错误,需要修改隐藏文件 .condarc文件的内容,具体路径为:c:\users\用户名\.condarc
使用记事本打开,复制下面这一段进去替换掉原本所有的内容,替换完之后保存即可
channels:
- defaults
show_channel_urls: true
channel_alias: http://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
此时再次输入命令新建yolov8环境,本文所用为python3.9,同时本教程所用pytorch源及python版本只支持3.9。
conda create -n yolov8 python=3.9
创建完之后输入
conda activate yolov8
进入yolov8环境之中,此时有NVIDIA显卡安装GPU版看3.3,其它需要安装CPU版看3.4。
3.3 安装GPU版本
3.3.1 安装pytorch(GPU版)
打开prompt 输入conda activate yolov8进入yolov8环境,之后输入下方命令即可安装pytorch,耐心等待安装完成(需要注意关闭加速软件否则会下载失败)
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
出现ERROR: No matching distribution found for torch==2.0.0+cu118等是因为python版本不对!
torch及相关库比较大,需要耐心等待下载完之后出现 done 则安装完成,因为是外网,如果下载不下来,可以点此下载torch 夸克网盘下载,然后通过pip install安装本地文件的方式安装,可以参考下图的方式,将torch放入d盘的test文件夹下,通过cd将文件目录跳转,再输入如下指令。(只支持python3.9版本,其余版本无法安装)
pip install torch-2.0.0+cu118-cp39-cp39-win_amd64.whl
耐心等待安装完成后,再输入下面的指令通过pip安装其余库。
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
3.3.2 安装ultralytics库
这个库包含了几乎全部运行yolov8所需的环境,输入以下命令后耐心等待即可
pip install ultralytics
如果报错 ERROR: No matching distribution found for ultralytics 则输入以下指令安装
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
也可以选择安装ultralytics的对应版本所需库,如pip install ultralytics==8.2.10,上方安装不报错则默认不需要安装指定版本。安装完成出现下图则环境配置完成。
3.4 安装CPU版本
3.4.1 安装pytorch(CPU版)
有英伟达显卡的按照上述流程安装后即可跳过本部分,无英伟达显卡则需要按照下述方法进行,
打开prompt ,进入yolov8环境之后,输入如下命令即可安装cpu版本的pytorch
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cpu
输入指令后看到下图再按下回车则开始正常安装。
耐心等待安装完成之后安装ultralytics库。
4.4.2 安装ultralytics库
pip install ultralytics
如果报错 ERROR: No matching distribution found for ultralytics 则输入以下指令安装
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
也可以选择安装ultralytics的对应版本所需库,如pip install ultralytics==8.2.10,上方安装不报错则默认不需要安装指定版本。安装完成出现下图则环境配置完成。
4. 下载yolov8源码
yolov8源码地址:GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLiteNEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite - ultralytics/ultralyticshttps://github.com/ultralytics/ultralytics/https://github.com/ultralytics/ultralytics/点进去下载代码,这里使用v8.2.0版本演示,进不去需要科学上网,或者点击下载 夸克网盘下载 ,我上传了多版本可以全部转存避免后续找不到,cat图片一并上传,下载8.3.0以下版本,(压缩包内附带yolov8n.pt、yolov8s.pt和yolov8m.pt预训练权重,链接资源失效请评论区反馈,看到会补,或者至公众号下载)可以下载下图所示几个预训练权重文件,常规使用yolov8n.pt即可。
下载完成之后解压到D盘或其它盘文件夹内,此时点击鼠标右键文件夹通过pycharm打开,打开后需要配置虚拟环境,点击file-settings,点击project:ultralytics,点击python interpreter,点击右边add....,若是中文则自行翻译后点击对应含义的位置。
点击Virtualenv Environment - Existing,点击右边三个点,找到刚才设置的D:\Anaconda3\envs\yolov8\python.exe 路径,之后一直点确定即可,耐心等待环境加载完毕。
或者新版本的界面更为简单,按照下图方式选择就可以
选择python,接着选择刚刚添加的yolov11环境,选择python.exe,
点击蓝字下载,建议yolov8n.pt速度相对快,下载过则跳过。
5. 验证环境
下载完成后将文件复制到ultralytics根目录下,即本文的ultralytics-8.2.0目录下,此时可以去网上下载一只猫猫的jpg图片(或者其它图片,根据自己喜好更改),修改文件名为cat.jpg。运行下面的指令需要cd到ultralytics-8.2.0目录,不然会报错。
检测环境是否有问题可以在prompt里yolov8环境下运行
yolo predict model=yolov8n.pt source='cat.jpg'
输出下图即环境正常
此时若报错 ModuleNotFoundError: No module named ‘hub_sdk‘
只需要 pip install hub_sdk 即可,安装完之后再次运行上述代码
如果报错 ImportError: DLL load failed while importing _imaging: 找不到指定的模块,是pillow版本问题导致的错误。输入如下代码更换pillow版本。
pip install pillow==8.4.0
如果报错中有numpy相关的语句,如 RuntimeError: Numpy is not available ;compiled using NumPy 1.x cannot be run in NumPy 2.1.2 as it may crash. To support both 1.x and 2.x versions of NumPy, modules must be compiled with NumPy 2.0. 等等,可以将numpy降级,输入如下代码即可解决此类问题。
pip install numpy==1.26.3
或者通过pycharm运行检测,新建一个yolov8_predict.py,在文件中添加如下代码。
from ultralytics import YOLO
# 加载预训练的 YOLOv8n 模型
model = YOLO('yolov8n.pt')
# 定义图像文件的路径
source = 'cat.jpg' #更改为自己的图片路径
# 运行推理,并附加参数
model.predict(source, save=True)
运行之后如下即环境正常
可以顺着路径查看检测后的图片,可爱的猫猫一张~
需要训练自己的数据集可以看这篇博客
如果遇到报错或者有疑问可以评论区交流,报错了先确定是按照我的教程配置的,仍然有问题再问。 关注微信公众号 快速联系我~