当前位置: 首页 > article >正文

最新yolov8环境搭建、推理训练一站式超详细教学

1、获取yolov8源码

访问yolov8_github官网,网络不稳定时可能需要加速器。yolov8源码地址
在这里插入图片描述
获取方式:直接下载或者git工具克隆
我使用git操作进行演示,复制github上的地址(需提前关闭加速器)。

git clone https://github.com/ultralytics/ultralytics.git

在这里插入图片描述
最终得到yolov8的源码
在这里插入图片描述

2、安装cuda、cudnn、anaconda、pytorch

Win+R输入cmd命令弹出对话框后输入命令,查看显卡驱动最高支持的CUDA版本
版本过低需要更新显卡驱动地址

nvidia-smi

在这里插入图片描述

安装cuda、cudnn(免邮箱注册)、pytorch统一以11.8版本进行安装

2.1、CUDA下载地址,点击CUDA Toolkit 11.8.0跳转页面后依次选择
在这里插入图片描述
下载完成后默认安装路径一直点确认即可
安装完成后,右击此电脑—>属性—>(界面右边)高级系统设置—>环境变量—>点击系统变量里的path进去查看环境变量。需要有下面两个环境变量,若没有安装默认安装路径自行添加
在这里插入图片描述
Win+R输入cmd命令弹出对话框后输入命令,验证cuda是否安装成功

nvcc -V

在这里插入图片描述

2.2、cudnn下载网址,选择对应11.x的版本即可,展开后鼠标点击安装包右键(不然会进行一系列邮箱注册繁琐操作)
如复制链接没有弹窗下载或者还是显示注册邮箱换一个11.x对应的cudnn版本即可
复制链接可直接用于迅雷下载速度更佳
请添加图片描述

下载解压后将所有文件拷贝到cuda目录里,cuda默认路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
在这里插入图片描述
拷贝后cuda中的文件为
在这里插入图片描述
2.3、安装anaconda、pytorch虚拟环境
2.3.1、anaconda教程很多,自行下载安装即可。安装完成后需要根据自己的安装路径配置环境变量
安装完成后,右击此电脑—>属性—>(界面右边)高级系统设置—>环境变量—>点击系统变量里的path进去添加环境变量。
在这里插入图片描述
找到自己安装的对应路径,复制即可:在这里插入图片描述
Win+R输入cmd命令弹出对话框后输入命令,验证Anaconda是否安装成功

conda --version
conda info
python
exit()

在这里插入图片描述
2.3.2、打开开始菜单,找到anaconda prompt
(1) 基于base创建一个命名为yolov8的python环境。-n 参数后跟新建环境名称

conda create -n yolov8 python==3.8

输入"y"即可,最后会得到如何激活环境的命令的界面
在这里插入图片描述

(2) 查看当前有那些虚拟环境

conda env list

在这里插入图片描述
(3)激活环境

conda activate yolov8

在这里插入图片描述
(4)pytorch官网地址,找到cuda11.8对应的命令进行安装
在这里插入图片描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述
等待安装完成,显示下图字样,即代表安装成功
在这里插入图片描述
找到源码对应文件,在文件路径处输入cmd,激活之前创建的yolo环境
在这里插入图片描述
安装requirements里的依赖

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

在这里插入图片描述

3、添加环境到pyCharm进行推理

打开PyCharm后,将现有yolov8环境添加到解释器中
在这里插入图片描述
然后创建一个test.py文件,检测pytorch

# 输入库
import torch
# 查看版本
print(torch.__version__)

# 查看gpu是否可用
print(torch.cuda.is_available())

# 查看对应CUDA的版本号
print(torch.backends.cudnn.version())
print(torch.version.cuda)

# 退出python
quit()

在这里插入图片描述
进行推理

yolo task=detect mode=predict model=yolov8n.pt conf=0.25 source='ultralytics/assets/bus.jpg'

若出现以下错误'yolo' 不是内部或外部命令,也不是可运行的程序或批处理文件
输入下面指令即可

pip install ultralytics

最后进行推理成功

在这里插入图片描述

未待完续…
总结:
1、获取github上yolov8的项目,利用git工具克隆到本地
2、下载cuda
3、下载cudnn
4、anaconda
5、创建yolov8虚拟环境
6、安装pytorch


http://www.kler.cn/a/144478.html

相关文章:

  • Vue3进阶5个小知识点 附带源码
  • jpom学习
  • Linux的基本指令(二)
  • Gen-2重磅发布“运动画笔”,可制作视频连环画啦!
  • 全面(16万字)深入探索深度学习:基础原理到经典模型网络的全面解析
  • springboot2.0 集成swagger3+Knife4j导出离线API 配置
  • 随笔记录-springmvc_ResourceHandlerRegistry+ResourceHttpRequestHandler
  • 【学习笔记】GameFramework的非官方实例TowerDefense-GameFramework-Demo的流程
  • BTS-GAN:基于MRI和条件对抗性网络的乳腺肿瘤计算机辅助分割系统
  • 【Pytorch】Visualization of Fature Maps(2)
  • RESTful
  • 3.读取字符串【2023.11.25】
  • Django(九、cookie与session)
  • laravel8安装多应用多模块(笔记三)
  • 数据库基础入门 — SQL排序与分页
  • 计数问题+约瑟夫问题(map)
  • 手写模拟Spring底层原理(Spring启动流程)(思维导图)
  • SQL 注入漏洞的理解
  • Python爬虫知识储备
  • AT89S52单片机智能寻迹小车自动红外避障趋光检测发声发光设计