YOLOv8项目推理从CPU到GPU

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

1.运行测试

运行E:\AI\yolo\yolov8\ultralytics-main\ultralytics\yolo\v8\detect\predict.py
YOLOv8项目推理从CPU到GPU
结果如下图,用CPU进行推理。

YOLOv8项目推理从CPU到GPU

2.查看Pytorch版本

进入yolov8虚拟环境:conda activate yolov8
查看Pytorch版本:pip list
YOLOv8项目推理从CPU到GPU
YOLOv8项目推理从CPU到GPU

3.安装CUDA

Nvidia显卡可用,ATI等其它显卡直接跳过划走!!!
查看显卡支持的CUDA版本:nvidia-smi
YOLOv8项目推理从CPU到GPU
我的电脑显卡最高支持CUDA Version:11.0
所以我安装CUDA11.0
CUDA下载链接
YOLOv8项目推理从CPU到GPU
根据自己的操作系统、构架、版本、安装方式选择。
YOLOv8项目推理从CPU到GPU

4.安装cuDNN

cuDNN下载链接
YOLOv8项目推理从CPU到GPU
解压出来,复制其中bin、include和lib,粘贴到CUDA的安装目录中。
YOLOv8项目推理从CPU到GPU
YOLOv8项目推理从CPU到GPU

5.安装PyTorch

进入PyTorch链接
选择带有+cu的命令
YOLOv8项目推理从CPU到GPU
使用安装命令

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

YOLOv8项目推理从CPU到GPU

查看Pytorch版本:pip list
YOLOv8项目推理从CPU到GPU

##6.运行测试
运行E:\AI\yolo\yolov8\ultralytics-main\ultralytics\yolo\v8\detect\predict.py
YOLOv8项目推理从CPU到GPU
是GPU了吧!
YOLOv8项目推理从CPU到GPU

7.查看结果

YOLOv8项目推理从CPU到GPU
YOLOv8项目推理从CPU到GPU
YOLOv8项目推理从CPU到GPU
搞定!!!

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年3月18日
下一篇 2023年3月18日

相关推荐