加强网站微信信息编辑队伍建设,各引擎收录查询,国内网站建设阿里云,wordpress可以注册了Yolov8的复现流程主要包含环境配置、下载源码和验证环境三大步骤#xff1a; 环境配置 查看电脑状况#xff1a;通过任务管理器查看电脑是否有独立显卡#xff08;NVIDIA卡#xff09;。若有#xff0c;后续可安装GPU版本的pytorch以加速训练#xff1b;若没有#xff0…Yolov8的复现流程主要包含环境配置、下载源码和验证环境三大步骤 环境配置 查看电脑状况通过任务管理器查看电脑是否有独立显卡NVIDIA卡。若有后续可安装GPU版本的pytorch以加速训练若没有则安装CPU版本不过训练速度会较慢。安装所需软件 Anaconda3安装可从清华镜像源选择对应系统版本的Anaconda3安装包下载。安装时选择“Just Me”安装路径建议为非C盘的纯英文路径如“D:\Anaconda3” 。安装过程中记得勾选添加到环境变量的选项推荐全选相关配置选项。Pycharm安装在官网下载社区免费版下载后打开安装程序选择安装路径建议非C盘 安装选项中勾选创建桌面快捷方式、更新PATH变量等然后点击安装。 安装环境 安装cuda及cudnn安装pytorch前需安装cuda。通过“nvidia - smi”命令查看显卡支持的CUDA最高版本若版本不足需更新显卡驱动。在NVIDIA Developer官网或提供的网盘链接下载匹配版本的cuda安装时建议选默认路径安装过程中选择自定义安装并全选组件。安装完成后用“nvcc - V”命令检查是否安装成功。之后进入cudnn官网下载与cuda版本匹配的cudnn Windows版本压缩包解压后将里面的三个文件夹内容复制到cuda的安装目录下进行覆盖替换。 创建虚拟环境打开Anaconda Prompt修改pip源和conda源为中科大源以加速下载。然后创建名为yolov8、python版本为3.9的虚拟环境若使用教程指定的pytorch源及版本python版本需为3.9命令为“conda create -n yolov8 python 3.9” 创建完成后用“conda activate yolov8”命令进入该环境。 安装GPU版本有NVIDIA显卡在yolov8环境中使用命令“pip install torch2.0.0cu118 torchvision0.15.1cu118 --extra - index - url https://download.pytorch.org/whl/cu118”安装pytorch的GPU版本。若下载失败可从夸克网盘下载对应版本的torch文件通过“pip install”安装本地文件的方式安装。安装完成后使用“pip install ultralytics”命令安装ultralytics库若安装报错可使用“pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple”命令安装。 安装CPU版本无NVIDIA显卡在yolov8环境中使用命令“pip install torch2.0.0 torchvision0.15.1 torchaudio2.0.1 --index - url https://download.pytorch.org/whl/cpu”安装pytorch的CPU版本。安装完成后同样使用“pip install ultralytics”命令安装ultralytics库报错处理方式与GPU版本安装时相同。 下载yolov8源码可从GitHub仓库“https://github.com/ultralytics/ultralytics/”下载yolov8源码也可从夸克网盘下载。 验证环境将下载的预训练权重文件复制到ultralytics根目录如ultralytics - 8.2.0目录下从网上找一张图片如猫猫图片命名为cat.jpg并放入相同目录。在yolov8环境下通过命令行运行“yolo predict modelyolov8n.pt source‘cat.jpg’” 或者在Pycharm中新建yolov8_predict.py文件添加如下代码运行验证
from ultralytics import YOLO
# 加载预训练的 YOLOv8n 模型
model YOLO(yolov8n.pt)
# 定义图像文件的路径
source cat.jpg
# 运行推理并附加参数
model.predict(source, saveTrue)运行后若出现类似 “image 1/1… 1 cat, …ms” 且结果保存路径正确的信息即表明环境正常。若报错根据报错信息进行相应处理如 “ModuleNotFoundError: No module named ‘hub_sdk‘” 错误执行 “pip install hub_sdk” “ImportError: DLL load failed while importing _imaging: 找不到指定的模块” 错误执行 “pip install pillow8.4.0” numpy相关错误执行 “pip install numpy1.26.3” 等。
最终运行结果
