跨境派

跨境派

跨境派,专注跨境行业新闻资讯、跨境电商知识分享!

当前位置:首页 > 综合服务 > 电商平台 > YOLOv8项目推理从CPU到GPU

YOLOv8项目推理从CPU到GPU

时间:2024-04-19 16:15:28 来源:网络cs 作者:淼淼 栏目:电商平台 阅读:

标签: 项目  推理 

YOLOv8项目推理从CPU到GPU

1.运行测试2.查看Pytorch版本3.安装CUDA4.安装cuDNN5.安装PyTorch7.查看结果

#YOLOv8项目推理从CPU到GPU
YOLOv8入坑出坑,Nvidia显卡可用,ATI等其它显卡直接跳过划走!!!
接YOLOv8代码调试运行实战

1.运行测试

运行E:\AI\yolo\yolov8\ultralytics-main\ultralytics\yolo\v8\detect\predict.py
在这里插入图片描述
结果如下图,用CPU进行推理。

在这里插入图片描述

2.查看Pytorch版本

进入yolov8虚拟环境:conda activate yolov8
查看Pytorch版本:pip list
在这里插入图片描述
在这里插入图片描述

3.安装CUDA

Nvidia显卡可用,ATI等其它显卡直接跳过划走!!!
查看显卡支持的CUDA版本:nvidia-smi
在这里插入图片描述
我的电脑显卡最高支持CUDA Version:11.0
所以我安装CUDA11.0
CUDA下载链接
在这里插入图片描述
根据自己的操作系统、构架、版本、安装方式选择。
在这里插入图片描述
下载的同时卸载掉所有NIVIDIA软件,然后再安装。具体安装过程网上太多了,忽略了。但是一定要记住自己的安装目录,下面要用到。
在这里插入图片描述
安装完成后,测试命令:nvcc -V

4.安装cuDNN

cuDNN下载链接
在这里插入图片描述
解压出来,复制其中bin、include和lib,粘贴到CUDA的安装目录中。
在这里插入图片描述
在这里插入图片描述

5.安装PyTorch

进入PyTorch链接
选择带有+cu的命令
在这里插入图片描述
使用安装命令

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述

查看Pytorch版本:pip list
在这里插入图片描述

##6.运行测试
运行E:\AI\yolo\yolov8\ultralytics-main\ultralytics\yolo\v8\detect\predict.py
在这里插入图片描述
是GPU了吧!
在这里插入图片描述

7.查看结果

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
搞定!!!

本文链接:https://www.kjpai.cn/news/2024-04-19/160386.html,文章来源:网络cs,作者:淼淼,版权归作者所有,如需转载请注明来源和作者,否则将追究法律责任!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。

文章评论