系列文章目录
文章目录
前言
提示:这里可以添加本文要记录的大概内容:
yolov8搭建(Windows,当前使用版本:8.2.48)
提示:以下是本篇文章正文内容,下面案例可供参考
一、项目地址
1.传送门
项目地址:
https://github.com/ultralytics/ultralytics
权重文件地址:
https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt
https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s.pt
https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m.pt
https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l.pt
https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x.pt
选择上述权重文件任意一个,放到项目根目录中(此处我使用yolov8n.pt)
二、CUDA配置
1.检查当前电脑允许的最大CUDA版本
代码如下(示例):执行命令:nvidia-smi 查看当前显卡支持的最大的cuda版本
2.下载并安装CUDA
下载地址:https://developer.nvidia.com/cuda-toolkit-archive(此处我选的是CUDA Toolkit 12.1.0,截至文章发表时间,此版本为最佳版本,需要与下面的pytorch版本匹配才可)
安装完毕后检查系统环境变量配置:
Path–>C:\Program File