首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >YOLOv8 入门指南:(1)环境配置

YOLOv8 入门指南:(1)环境配置

作者头像
张高兴
发布2025-05-21 13:51:40
发布2025-05-21 13:51:40
96600
代码可运行
举报
文章被收录于专栏:张高兴的博客张高兴的博客
运行总次数:0
代码可运行

什么是 YOLO

YOLO(You Only Look Once)模型是一种基于深度学习的目标检测算法,它可以在一张图片中识别出多个物体的类别和位置。YOLO 的特点是速度快,准确率高,而且只需要看一次图片就可以做出预测,不需要像其他算法那样先提取候选区域再进行分类。YOLO 的原理是将图片划分为网格,每个网格预测一定数量的边界框和类别概率,然后根据置信度和非极大值抑制筛选出最终的检测结果。YOLO 目前有多个版本,最新的是 YOLOv8,在2023年1月10日开源,提供了更多的模型选择和更好的性能。

确定环境版本

  1. 确定 CUDA、cuDNN 版本; 在 NVIDIA 控制面板中点击帮助-系统信息-组件中查看显卡支持的最高版本。
  1. 确定 PyTorch 支持的 CUDA 最高版本。 进入 PyTorch 官网。

安装 CUDA Tookit

CUDA 是建立在 NVIDIA GPU 上的一个通用并行计算平台和编程模型,它提供了 GPU 编程的简易接口,基于 CUDA 编程可以构建基于GPU计算的应用程序,利用 GPU 的并行计算引擎来更加高效地解决比较复杂的计算难题。

进入 CUDA 下载页,找到合适的版本下载;

下载完成后,运行 exe 文件安装即可;

验证安装结果,在命令行中运行命令,查看 CUDA 的信息:

代码语言:javascript
代码运行次数:0
运行
复制
nvcc -V

安装 cuDNN

cuDNN 是一个深度学习加速库,提供了一套标准的 API,通过 cuDNN 来加速深度学习框架在 NVIDIA GPU 上的运算。

  1. 注册 NVIDIA 开发者账号并登录;
  2. 进入 cuDNN 下载页,找到对应的 CUDA 版本下载;
  1. 解压复制到 CUDA 安装的文件夹中。

安装 PyTorch

PyTorch 是一个开源的 Python 机器学习框架,YOLO 使用了 PyTorch 框架。

在命令行中运行命令,更新 pip 镜像源:

代码语言:javascript
代码运行次数:0
运行
复制
pip config set global.index-url https://mirror.nju.edu.cn/pypi/web/simple

进入 PyTorch 官网;

复制安装命令:

代码语言:javascript
代码运行次数:0
运行
复制
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

在命令行中运行命令。

验证

在 Python 环境中,输入下面三行,看打印结果

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2024-02-29,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 土味儿编程 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 确定环境版本
  • 安装 CUDA Tookit
  • 安装 cuDNN
  • 安装 PyTorch
  • 验证
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档