YOLO V8语义分割模型部署

目录

[1 配置pytorch环境](#1 配置pytorch环境)

[2 配置yolo环境](#2 配置yolo环境)

[3 测试yoloV8的语义分割模型](#3 测试yoloV8的语义分割模型)


1 配置pytorch环境

我的电脑为Y9000P 4090,英伟达显卡驱动版本为525.105.17,驱动显示最高的cuda版本号为12.0,cuda版本为11.6,cudnn版本号为8.5.0。Anaconda3版本为22.9.0。如下:

对于跑yolo来说,我们创建python3.8环境。

复制代码
conda create -n yo python=3.8

启动yo环境:

复制代码
conda activate yo

我的cuda是11.6,安装pytorch1.13.0并用pip加速。

复制代码
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

安装成功,测试cuda是否可用:

复制代码
(yo) liuhongwei@liuhongwei-Legion-Y9000P-IRX8H:~$ python
Python 3.8.18 | packaged by conda-forge | (default, Oct 10 2023, 15:44:36) 
[GCC 12.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cud
torch.cuda                          torch.cudnn_convolution_relu(
torch.cudnn_affine_grid_generator(  torch.cudnn_convolution_transpose(
torch.cudnn_batch_norm(             torch.cudnn_grid_sampler(
torch.cudnn_convolution(            torch.cudnn_is_acceptable(
torch.cudnn_convolution_add_relu(   
>>> torch.cud
torch.cuda                          torch.cudnn_convolution_relu(
torch.cudnn_affine_grid_generator(  torch.cudnn_convolution_transpose(
torch.cudnn_batch_norm(             torch.cudnn_grid_sampler(
torch.cudnn_convolution(            torch.cudnn_is_acceptable(
torch.cudnn_convolution_add_relu(   
>>> torch.cuda.is_available()
True
>>> 

这样就是成功的了。

2 配置yolo环境

yoloV8环境及其好配置,只需输入一行命令即可安装所有所需依赖:

复制代码
pip install ultralytics

安装完成。

从官网下载yolo:

yoloV8官网https://github.com/ultralytics/ultralytics 或者直接克隆到桌面:

复制代码
git clone https://github.com/ultralytics/ultralytics.git

下载预训练模型:在它的官方网站上有用作各种任务的预训练权重。

我们下载目标检测和语义分割权重:

将这些权重放到yoloV8目录下,创建权重文件夹weights,将权重放入其中:

目录结构如下:

至此,我们配置好了yoloV8,开始进行测试。

3 测试yoloV8的语义分割模型

我们在网上随便下载一张图像:

复制代码
yolo task=segment mode=predict model=weights/yolov8s-seg.pt source=0

预测,可能会出错,不用慌,教程没有问题,是你的conda环境有问题,撞库了。

fusedconv.weight.copy_(torch.mm(w_bn, w_conv).view(fusedconv.weight.shape))

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`

解决办法:

清空运行时库缓存:

bash 复制代码
unset LD_LIBRARY_PATH

运行成功,我们把source换成我们自己的图像:

成功

相关推荐
羊羊小栈3 天前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业
大霸王龙4 天前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘
m_136874 天前
Mac M 系列芯片 YOLOv8 部署教程(CPU/Metal 后端一键安装)
yolo·macos
格林威4 天前
机器视觉在半导体制造中有哪些检测应用
人工智能·数码相机·yolo·计算机视觉·视觉检测·制造·相机
、、、、南山小雨、、、、5 天前
YOLO在ubuntu22安装
yolo
羊羊小栈5 天前
基于「YOLO目标检测 + 多模态AI分析」的铁路轨道缺陷检测安全系统(vue+flask+数据集+模型训练)
人工智能·yolo·目标检测·语言模型·毕业设计·创业创新·大作业
Python图像识别6 天前
63_基于深度学习的草莓病害检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
范男6 天前
YOLO11目标检测运行推理简约GUI界面
图像处理·人工智能·yolo·计算机视觉·视觉检测
model20056 天前
ubuntu24.04+5070ti训练yolo模型(2)
人工智能·yolo
强盛小灵通专卖员6 天前
RK3576边缘计算设备部署YOLOv11
人工智能·深度学习·yolo·边缘计算·sci·rk3576·小论文