【深度学习之YOLO8】环境部署_yolov8部署

本文详细介绍了如何搭建YOLO8的深度学习环境,包括确定CUDA和cuDNN版本、安装Python和Anaconda、配置CUDA toolkit、安装cuDNN以及PyTorch。重点强调了版本兼容性和环境变量的设置,确保每个步骤都经过验证。
摘要由CSDN通过智能技术生成

配置Python环境变量

创建conda环境

配置CUDA环境变量

配置cuDNN文件

下载安装PythonSDK

拉取yolo8 GitHub项目到本地

Anaconda安装

下载安装CUDA toolkit

下载PyTorch

下载cuDNN

环境测试

除了拉取代码不需要验证,其他都需要自己check下到底是不是真安装成功了

一、确定版本

CUDA toolkit、cuDNN版本

  1. 查看显卡的CUDA支持的最高版本,我的是12.2.79,后面安装的CUDA toolkit和cuDNN大版本不能超过它
    在这里插入图片描述
  2. PyTorch官网,查看支持的CUDA最高版本,即:我的电脑torch是最高支持11.8的CUDA
    在这里插入图片描述

由以上两点,得出安装的CUDA、cuDNN不能超11.8,那我后面这俩安装<=11.8

Python、PyTorch版本

torch、python各版本兼容情况表

torch torchvision Python
main / nightly main / nightly >=3.8, <=3.11
2.0 0.15 >=3.8, <=3.11
1.13 0.14 >=3.7.2, <=3.10
1.12 0.13 >=3.7, <=3.10
1.11 0.12 >=3.7, <=3.10
1.10 0.11 >=3.6, <=3.9
1.9 0.10 >=3.6, <=3.9
1.8 0.9 >=3.6, <=3.9
1.7 0.8 >=3.6, <=3.9
1.6 0.7 >=3.6, <=3.8
1.5 0.6 >=3.5, <=3.8
1.4 0.5 ==2.7, >=3.5, <=3.8
1.3 0.4.2 / 0.4.3 ==2.7, >=3.5, <=3.7
1.2 0.4.1
你可以按照以下步骤来部署 YOLOv5 到 NCNN: 1. 首先,你需要克隆 YOLOv5 仓库到本地: ```bash git clone https://github.com/ultralytics/yolov5.git cd yolov5 ``` 2. 接下来,你需要安装依赖项: ```bash pip install -r requirements.txt ``` 3. 现在,你需要将 YOLOv5 模型转换为 ONNX 格式。运行以下命令: ```bash python models/export.py --weights yolov5s.pt --img-size 640 --batch-size 1 ``` 这将生成一个 `yolov5s.onnx` 文件。 4. 下一步是克隆 NCNN 仓库: ```bash git clone https://github.com/Tencent/ncnn.git ``` 5. 进入 `ncnn` 目录,然后创建一个新的目录来存放 YOLOv5 模型: ```bash cd ncnn mkdir models/yolov5 ``` 6. 将生成的 `yolov5s.onnx` 文件复制到 `ncnn/models/yolov5` 目录中。 7. 接下来,你需要修改 `examples/yolov5.cpp` 文件,以适应 YOLOv5 模型。你可以使用编辑器打开该文件,然后按照其中的注释进行修改。 8. 编译 NCNN。你需要按照 NCNN 的文档中的说明进行编译。根据你的平台和需求进行相应的设置。 9. 编译完成后,你可以使用以下命令运行 YOLOv5 在 NCNN 上: ```bash ./build/examples/yolov5/yolov5 models/yolov5/yolov5s.param models/yolov5/yolov5s.bin <image_path> ``` 这将在给定的图像上运行 YOLOv5,并输出检测结果。 请注意,这只是一个基本的指南来部署 YOLOv5 到 NCNN。具体的步骤可能因环境和需求而有所不同。你可能需要根据自己的情况进行调整和修改。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值