YOLOv5目标检测学习(2):运行一个yolo应用所需要配置的深度学习环境

news/2024/7/10 1:03:56 标签: YOLO, 目标检测, 学习

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录

  • 前言
  • 一、CUDA、CUDNN的下载安装
    • 1.1 CUDA的安装
    • 1.2 cuDNN的配置
  • 二、anaconda的安装
    • 1.卸载python
    • 2.安装anaconda
  • 三、Pytorch、python虚拟环境安装
    • 1.创建虚拟环境
    • 2.安装pytorch
  • 四、pycharm上运行深度学习示例
  • 五、总结所有软件的版本


前言

我想完整的记录一下一个新手接触深度学习和yolo算法完成目标检测任务的全过程,包括基本软件环境的下载安装、训练集的准备和处理、训练的过程、参数的调整、性能的分析,以及模型的部署这几个方面。

一、CUDA、CUDNN的下载安装

我的电脑配置:Windows10系统,10代i5,2060显卡,16G内存,安装有python3.8版本
在这里插入图片描述

软件:
①IDE:pycharm
②深度学习的模型训练工具:CUDA、CUDNN
③深度学习管理工具:anaconda

由于pycharm我已经安装好了,所以就不介绍了。

1.1 CUDA的安装

①打开英伟达控制界面,可以查看cuda支持的版本
在这里插入图片描述
这一步主要是为了看显卡支持cuda最高哪个版本,我的最高支持12.2.146版本
②下载CUDA toolkit

链接: cuda官网
我选择下载12.2.0版本
在这里插入图片描述
选择Windows版本的local本地下载模式。
在这里插入图片描述
然后一路同意即可。
其中选择自定义安装,可以安装到其他盘去,不用安C盘。
安装好后可以查看到版本号
在这里插入图片描述

还可以查看cuda的安装路径。
在这里插入图片描述
在这里插入图片描述
下载好后文件如图:
在这里插入图片描述

1.2 cuDNN的配置

链接: cudnn
这里首先注意一个问题,我下载的cuda是2023年6月的12.2版本,所以cudnn我选择了相匹配的版本
在这里插入图片描述
然后解压。下载后发现其实cudnn不是一个exe文件,而是一个压缩包,解压后,有三个文件夹,把三个文件夹拷贝到cuda的安装目录下。
cuDNN 其实就是 CUDA 的一个补丁而已,专为深度学习运算进行优化的。然后再参加环境变量。

下一步是添加环境变量。
在这里插入图片描述
就像这样添加,要把bin、lib、include、libnvvp都添加进去。
最后打开终端进行验证。输入这两个exe程序,只要最后是pass就可以了。
在这里插入图片描述
在这里插入图片描述
注意:cuda没有samples文件夹是因为11.6版本后就取消了,想要自己去git上下。

二、anaconda的安装

1.卸载python

安装anaconda前需要先卸载掉以前下载的python,并删除环境变量。
在这里插入图片描述
用终端查看:
在这里插入图片描述
现在没有python了,而且好的一点是环境变量里的python路径是自动删除的,不用手动去删了。

2.安装anaconda

使用清华镜像源下载,我选了一个不早也不新的版本。
在这里插入图片描述
在这里插入图片描述
最后如图:
在这里插入图片描述
输入下面的五个环境变量。
在这里插入图片描述

简要说明五条路径的用途:这五个环境变量中,1是Python需要,2是conda自带脚本,3是jupyter notebook动态库, 4是使用C with python的时候.
最后可以在终端里面验证。
在这里插入图片描述

下载好anaconda后,会自动安装一个新版本的python,我这里查看是3.11.4版本
在这里插入图片描述

三、Pytorch、python虚拟环境安装

为什么需要学会这个呢?其实刚刚安装好Anaconda会自带一个基础环境。但是我们后续的项目每一个需要的安装包不同,为了避免冲突,所以我们可以为每一个项目配置一个虚拟环境,这样就不相互打扰了。

1.创建虚拟环境

用anaconda终端:
在这里插入图片描述
输入这个命令:conda env list
来查看当前有多少虚拟环境。
在这里插入图片描述
刚好我这里只有一个base环境。
在这里插入图片描述
利用 conda create 指令创建新的虚拟环境

conda create –n 虚拟环境名字(如yolov5) python=版本

我这里是:

conda create –n yolov5p1 python=3.11.4

在这里插入图片描述
输入y,继续:
在这里插入图片描述
加载一堆东西后界面如图。
在这里插入图片描述
现在就有两个环境了。

2.安装pytorch

链接: pytorch
这个是pytorch官网,可以直接找到系统推荐的pytorch版本。
在这里插入图片描述
这里我遇到一个问题,就是这里pytorch支持的是cuda的11.8和12.1版本,但是我之前下载的cuda是12.2版本的,不知道后面会有什么影响。
然后复制下面的指令,放到anaconda终端里运行:
经过漫长的安装,最后如图:
在这里插入图片描述
会有各种各样的库。
输入python
输入 import torch
输入 torch.cuda.is_available(),这个命令是检查我们pytorch的GPU能否用。
如果显示True,就说明我们这个PyTorch安装成功了
在这里插入图片描述

四、pycharm上运行深度学习示例

这里奇怪的是,别人的教程里配置pycharm和pytorch还有好几步,但是这里我直接就可以运行代码:

import torch
print(torch.cuda.is_available())

打开pycharm,创建一个例程,运行结果如图:
在这里插入图片描述

至此,所有的深度学习环境就配置好了。

五、总结所有软件的版本

CUDA:12.2.0
cuDNN:8.9.2
anaconda:23.7.2
python:3.11.4


http://www.niftyadmin.cn/n/5413069.html

相关文章

任务调度新境界:探秘ScheduledExecutorService的异步魔力

欢迎来到我的博客,代码的世界里,每一行都是一个故事 任务调度新境界:探秘ScheduledExecutorService的异步魔力 前言ScheduledExecutorService的基本概念基本概念:为何它是 Java 中任务调度的首选工具:基本用法&#xf…

Unity Samples和帧动画的问题

拖动序列帧图片和自己创建clip的帧率不同 我今天在创建帧动画的时候用了两种方式第一种是直接拖动序列帧图片到Hierachy,然后生成的第二种是这样我发现两者播放的动画速率不一样最后查了半天查不到原因。最后发现是Samples的原因,而且Unity把Samples这个…

Debezium发布历史165

原文地址: https://debezium.io/blog/2023/10/05/Debezium-JMX-signaling-and-notifications/ 欢迎关注留言,我是收集整理小能手,工具翻译,仅供参考,笔芯笔芯. Debezium signaling and notifications - Part 3: JMX …

【Linux】常见的基本指令(上)

在这篇博客中,将会介绍到Linux操作系统的基本指令。 一.ls指令 ls 指令是显示当前目录的下文件和文件夹。 ls -a 列出目录下的所有文件,包括隐藏文件 ls -l 或者 ll 列出当前目录下的文件和文件夹,并且是列文件的…

前端技术研究越深入,越觉得技术不是决定录用唯一条件。

一、拒绝抬杠 我说技能不是唯一条件,不是说技能不重要,招聘前端条件是1X,其中1是技能,X是其他条件。 如果X条件很优秀,1这个条件可以降格为0.8、0.5,甚至更低。 有人就抬杠,那为啥不招聘清洁工来干前端&…

记一次busybox-mountPath简单但容易忽略的问题open /proc/self/fd: no such file or directory

目录 一.设备条件介绍 二.我的目的 三.问题所在 1.在用以下pod进行测试的时候出现问题 2.报错 3.解决 4.解决完成 一.设备条件介绍 [rootk8s-master pv]# containerd --version containerd containerd.io 1.6.25 d8f198a4ed8892c764191ef7b3b06d8a2eeb5c7f ​ [rootk…

基于YOLOv5的驾驶员疲劳驾驶行为​​​​​​​检测系统

💡💡💡本文主要内容:详细介绍了疲劳驾驶行为检测整个过程,从数据集到训练模型到结果可视化分析。 博主简介 AI小怪兽,YOLO骨灰级玩家,1)YOLOv5、v7、v8优化创新,轻松涨点和模型轻量…

腾讯云服务器和阿里云服务器价格测评_2024年费用大PK

2024年阿里云服务器和腾讯云服务器价格战已经打响,阿里云服务器优惠61元一年起,腾讯云服务器61元一年,2核2G3M、2核4G、4核8G、4核16G、8核16G、16核32G、16核64G等配置价格对比,阿腾云atengyun.com整理阿里云和腾讯云服务器详细配…