首页 > 其他分享 >YOLOv8超详细环境搭建以及模型训练(GPU版本)

YOLOv8超详细环境搭建以及模型训练(GPU版本)

时间:2024-08-25 09:22:02浏览次数:14  
标签:YOLOv8 点击 虚拟环境 CUDA GPU yolov8 安装 下载 搭建

目录

1.安装CUDA和cuDNN

1.1安装CUDA

1.1.1查看当前你的电脑显卡支持的最高CUDA版本,后面的安装不能超过它

1.1.2 下载CUDA(官网或者百度网盘)

1.1.3 安装CUDA 11.8

1.2配置cuDNN

1.2.1下载cuDNN(官网或者百度网盘)

1.2.2 配置cuDNN

2.安装Anaconda

2.1 下载Anaconda

2.2 安装Anaconda

2.3 配置环境变量

3.创建python虚拟环境

3.1创建python虚拟环境

3.2 虚拟环境的路径设置( 常见问题以及解决方法)

4.打开项目

4.1 然后打开yolov8项目

5.在Pycharm中调用yolov8环境

6.pycharm中添加镜像地址

7.安装GPU版本的torch

8.安装其他的项目依赖

9.以训练NEU-DET数据集为例训练模型

9.1创建数据集

9.2 创建数据集的配置文件NEU-DET.yaml文件

9.3 创建模型对象yolov8-neu-det.yaml

9.5模型训练的成功截图

-----------------------------------------------------------------------------------------------------------

1.安装CUDA和cuDNN

1.1安装CUDA

1.1.1查看当前你的电脑显卡支持的最高CUDA版本,后面的安装不能超过它

方法一:通过命令的方式查看

nvidia-smi

方法二:【鼠标右键桌面】--【显示更多选项】--【NVIDIA控制面板】--在控制面板的左下角点击【系统信息】--选择【组件】

NVIDIA CUDA 12.1.68 driver就是显卡支持的CUDA版本

1.1.2 下载CUDA(官网或者百度网盘)

(1)下载地址一(官网):

下载小于12.1版本都都可以,我这里下载的是11.8.0,然后点击【CUDA Toolkit 11.8.0】

  • 下载CUDA11.8

(2)下载地址二(百度网盘):

11.8.0版本的百度网盘下载地址,链接:百度网盘 请输入提取码

提取码:h0tw

1.1.3 安装CUDA 11.8

  • 点击刚下载的CUDA进行安装

  • 下面的路径是提取地址,安装完后会自动删除,选择默认即可,点击【OK】

  • 等待提取完成

  • 点击【同意并继续】

  • 选择【自定义】,点击【下一步】

  • 驱动组件可以全部勾选,点击【下一步】

  • cmd命令【nvcc -V】查看CUDA是否安装成功,如果出现如下界面表示CUDA安装成功

1.2配置cuDNN

1.2.1下载cuDNN(官网或者百度网盘)

(1)下载地址一(官网)

下载cuDNN,先登录,按照提示流程注册、登录即可。

cuDNN选择支持CUDA的版本(刚刚下载的CUDA是哪个版本就选择对应的版本)(CUDA 11.x则代表支持CUDA11代版本的所有版本)

下载对应自己电脑系统的版本。

(2)下载地址一(百度网盘)

链接:百度网盘 请输入提取码

提取码:bwas

1.2.2 配置cuDNN

解压cuDNN压缩包,然后将cuDNN文件夹里面的几个文件复制到CUDA的安装目录下直接覆盖

覆盖

2.安装Anaconda

2.1 下载Anaconda

(1)Anaconda下载方式一(官网)

Anaconda下载官网:Anaconda | The Operating System for AI

点击下载【Free Download】即可

然后点击【Download】

(2)Anaconda下载方式二(百度网盘)

链接:百度网盘 请输入提取码

提取码:9hcx

2.2 安装Anaconda

点击安装包,然后点击【Next]

  • 点击【I Agree】

  • 点击【Next】

  • 安装路径的选择,点击【Next】

  • 点击【INstall】,然后等待安装完成

  • 点击【Next】

  • 点击【Next】

  • 点击【Finish】

2.3 配置环境变量

右键点击【我的电脑】(此电脑)--【属性】--点击【高级系统设置】--【高级】--【环境变量】-找到系统变量下的【Path】--右侧点击【新建】,按照我标红框的进行添加即可。(注意anaconda3的安装路径

3.创建python虚拟环境

3.1创建python虚拟环境

虚拟环境的作用:虚拟环境可以隔离不同项目的依赖项,避免不同项目之间的干扰,使项目更加稳定。

首先进入命令界面

输入如下命令:

conda create -n yolov8 python=3.11.4

其中:yolov8是虚拟环境名称(可以任意取名),python=3.11.4是指定python的版本(也可指定另外的版本)

输入【y】,之后回车开始安装

如果出现如下界面,表示安装完成。

输入命令可以查看虚拟环境否创建成功:conda env list

如下界面表示创建成功(虚拟环境名称以及虚拟环境的路径):

激活进入环境:activate yolov8

激活成功后,前面的base会替换成yolov8

备注:如果需要删除刚刚创建的虚拟环境,可以通过如下命令删除:

conda remove -n yolov8 --all

3.2 虚拟环境的路径设置( 常见问题以及解决方法)

(如果3.1在创建虚拟环境的时候,路径不是anaconda3的安装路径下,而是总是在c盘,才需要进行3.2小节的设置,否则跳过3.2小节)

解决方法参考:【BUG】使用Anaconda3创建虚拟环境时,总是安装到C盘全局默认路径,而不是安装到自定义的Anaconda的envs目录下的解决方法-CSDN博客

4.打开项目

首先下载YOLOv8项目

(YOLO系列项目官网下载地址:主页 -Ultralytics YOLO 文档

yolov8官网下载地址:https://github.com/ultralytics/ultralytics

yolov8百度网盘下载地址链接:百度网盘 请输入提取码

提取码:8t0c

yolov8压缩包下载完成之后,解压到某个路径下(可以解压在D盘)

4.1 打开yolov8项目

Pycharm前提已经安装完成(如果没有安装,可以网上搜索Pycharm的安装教程,有很多)

Pycharm软件中打开刚解压的yolov8项目

5.在Pycharm中调用yolov8环境

点击【File】-->【setting】-->【Project】-->【Python Interpreter】-->【Add Interpreter】-->【Add Local Interpreter】

设置虚拟环境

备注:在Pycharm中如何查看当前项目使用的虚拟环境:

点击左下角的【Terminal】,可以看出使用的虚拟环境

或者在当前项目下新建一个test-p.py文件,输入如下代码并运行

import sys
print(sys.prefix)

截图如下,可以看出使用的是yolov8虚拟环境以及虚拟环境的路径:

6.pycharm中添加镜像地址

添加镜像可以加快下载依赖包的速度。

点击图标【Python Packages】--点击设置按钮【Manage respositories】

点击【+】--在【Name】处随意更改名字,然后在【Repository URL】填上镜像源地址,最后点击【OK】

也可以添加另外一些镜像地址:

1.清华大学镜像源:

https://pypi.tuna.tsinghua.edu.cn/simple/

2.阿里云镜像源:

http://mirrors.aliyun.com/pypi/simple/

3.中国科技大学镜像源:

https://pypi.mirrors.ustc.edu.cn/simple/

7.安装GPU版本的torch

torch官网:Previous PyTorch Versions | PyTorch

选择和CUDA对应的torch版本(前面自己已经安装的版本),然后复制

  • 进入pycharm,点击【Teminal】,进入yolov8虚拟环境之后,粘贴上面复杂的torch版本进行安装(然后等待安装完成)

8.安装其他的项目依赖

左下角终端打开,安装项目依赖,执行命令:pip install -r requirements.txt

如果没有requirements.txt这个文件,可以直接在项目下新建这个文件,将下面的内容复制进去,然后执行安装命令

certifi==2022.12.7
charset-normalizer==2.1.1
colorama==0.4.6
contourpy==1.2.0
cycler==0.12.1
filelock==3.9.0
fonttools==4.50.0
fsspec==2024.6.1
huggingface-hub==0.23.4
idna==3.4
Jinja2==3.1.2
kiwisolver==1.4.5
MarkupSafe==2.1.3
matplotlib==3.8.3
mpmath==1.3.0
networkx==3.2.1
numpy==1.26.3
opencv-python==4.9.0.80
packaging==24.0
pandas==2.2.1
pillow==10.2.0
psutil==5.9.8
py-cpuinfo==9.0.0
pyparsing==3.1.2
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML==6.0.1
requests==2.28.1
scipy==1.12.0
seaborn==0.13.2
six==1.16.0
sympy==1.12
thop==0.1.1.post2209072238
#torch==2.0.0+cu118
#torchaudio==2.0.1+cu118
#torchvision==0.15.1+cu118
tqdm==4.66.2
typing_extensions==4.8.0
tzdata==2024.1
ultralytics==8.1.34
urllib3==1.26.13

9.以训练NEU-DET数据集为例训练模型

9.1创建数据集

数据集下载地址:

链接:https://pan.baidu.com/s/1peB3kd6T7xNoKnjxsrKHuw?pwd=xqcm 
提取码:xqcm

然后将下载的数据集解压之后放在项目的data目录(如何没有此目录,自己创建一个)下。

9.2 创建数据集的配置文件NEU-DET.yaml

在项目的【ultralytics/cfg/datasets】目录下新建一个NEU-DET.yaml,内容如下

path: D:/YOLO/YOLOv8/data/NEU-DET # 注意修改自己NEU-DET数据集的路径
train: train/images  # train images
val: val/images  # val images
test: test/images  # test images

# Classes

names:
  0: crazing
  1: inclusion
  2: patches
  3: pitted_surface
  4: rolled-in_scale
  5: scratches

截图如下

9.3 创建模型对象yolov8-neu-det.yaml

在项目的【ultralytics/cfg/models/v8】目录下新建一个yolov8-neu-det.yaml,代码如下

# Ultralytics YOLO 

标签:YOLOv8,点击,虚拟环境,CUDA,GPU,yolov8,安装,下载,搭建
From: https://blog.csdn.net/2401_85556416/article/details/141394730

相关文章

  • 2024玩儿转TikTok之环境介绍及独立使用住宅ip搭建,最便宜的网络,20元/月,刚刚更新最新教
    郑重申明:本文章只对合法合理做tiktok视频运营的用户做学习交流使用,有其他使用不当的违规违法行为后果自负!网络的环境测试,必须是独立住宅ip,网络环境优良,tiktok发布作品才能有流量1、选择服务器我选择的服务器:https://ipraft.com/?i71b7cf​​​然后下载FinalShell连接......
  • 使用Cloudflare Worker搭建自己的AI绘画工具
    demo:https://aidraw.foxhank.top0.前言Cloudflare公司推出了workers-ai,可以免费在Cloudflare的全球网络上运行由无服务器GPU提供支持的机器学习模型。WorkersAI可以Cloudflare网络上使用自己的代码运行机器学习模型,也就是说,只要写一个js代码,就可以免费调用cloudfl......
  • 向量数据库Faiss的搭建与使用
    Faiss(FacebookAISimilaritySearch)是一个高效的向量相似性搜索库,专为高维向量数据的搜索和聚类设计。以下是如何搭建和使用Faiss的详细步骤。1.环境准备安装依赖首先,确保你的开发环境满足Faiss的要求。Faiss支持Python,假设你要使用Python版本。安装Anaconda或Miniconda(......
  • 怎么实现用frp搭建一个自己的内网穿透服务
    使用frp搭建一个自己的内网穿透服务包括以下几个步骤:配置frp服务器(服务端)和frp客户端。Frp是什么:frp(FastReverseProxy)是一款高性能的反向代理应用,广泛用于内网穿透、跨网络访问等场景。以下是frp的一些常见应用场景:1.内网服务的外网访问frp可以将内网中的Web......
  • 博客站搭建需求设计书
    前言该文章适合的读者需要自己创作博客网站,并参考他人个人博客网站的需求设计,该文章结尾处有优秀案例收集1.项目概述1.1项目背景创建个人博客网站的动机和目的个人技术能力展示,用于个人的技术文章,技术思维图,产品架构图,个人思维感悟,有趣生活展示,类似于独特的社交名片。1.2......
  • AD(Active Directory )域的搭建与操作
    AD域的搭建与操作一、准备工作准备好VM虚拟机和Server的安装包。二、安装Server2022选择标准且有图形界面的进行安装。选择自定义安装方式。为虚拟机server2022安装VMwaretools。回到桌面,右键个性化把计算机和网络图标放出来。三、安装AD域服务......
  • 第7篇:在虚拟机 centos7上搭建jira管理工具
    本文详细介绍了如何在CentOS7系统上下载配置Jira,包括创建文件夹、下载安装包、解压、修改配置文件以及设置JVM和MySQL环境。同时,文章还涉及了JDK1.8的安装,数据库的创建,以及Jira的破解步骤,包括替换特定jar文件和配置数据库连接。最后,文章提到了启动Jira服务并进行汉化包的下载链接......
  • 2.ros2创建话题+发布消息(VScode环境搭建+编写)
    前言:这是在1.ROS2安装-创建节点的基础上进行改进的,由于ubuntu自带的nano和vim在编写的时候不是很方便,在这教大家怎么用VScode进行编写代码1.Vscode安装1.1下面是Vscode的deb网盘连接,你也可以去官网下载        官网连接:DownloadVisualStudioCode-Mac,Linux,......
  • yolov8行人车辆检测与计数系统 (行人车辆跟踪计数)
     yolov8行人车辆检测与计数系统(Python+YOLOv8+deepsort车辆追踪深度学习模型+清新界面)(1)YOL v8算法实现,模型一键切换更新;(2)检测图片、视频等图像中的各目标数目;(3)摄像头监控实时检测,便携展示、记录和保存;(4)支持切换目标,各目标位置切换查看;(5)提供数据集和训练代码可重新训练;......
  • 利用DockerDesktop在Win10下搭建Gitlab
    前言之前尝试过直接通过DockerDesktop来搭建Gitlab,但是最终因为稳定性、数据安全、数据备份的问题,还是选择了在Linux服务器上搭建正式的gitlab,仅记录这样的一个过程。个人建议这里需要注意两点:1、操作系统的版本经过我个人尝试,如果是使用win10的话,建议使用1809版本以上的操作系......