yolov5环境GPU搭建 ,用GPU跑polov5算法

news2024/11/25 4:57:51

win10+NVIDIA GeForce RTX 3050+torch1.13.1+torchaudio0.13.1+torchvision 0.14.1 +cuda11.7+python3.8+cudnn8.7.0

在环境搭建中踩了许多坑,yolov5环境的搭建需要依赖很多环境,用cpu跑很容易跑单张识别,用GPU跑却踩了很多坑,不过GPU环境搭建起来后续训练数据集我想就会快很对!GPU搭建第一步别着急

1.查看电脑显卡

使用PowerShell

  1. 使用PowerShell命令
    • 按下 Windows 键,输入 PowerShell,然后右键选择“以管理员身份运行”。
    • 在PowerShell窗口中,输入以下命令并按回车:
      Get-WmiObject Win32_VideoController | Select-Object Name, Description, DriverVersion
      
      3.输入后显示我的显卡驱动为3050
    • 同时查看电脑cuda版本可以安装什么版本
    • 打开nvidia(桌面右键)->选择左下角的系统信息->组件

      第三行,可以看到自己电脑支持的cuda这里显示的是11.7

2环境依赖

yolov5依赖环境包含cuda ,cudnn,Anaconda,torch, torchvision,torchaudio,Anaconda下载好了就是powershell命令端,要在这个命令端口搭建虚拟环境

(1)cuda和cudnn安装省略:请参照这个博主的安装方法

CUDA安装教程(超详细)-CSDN博客,他安装的版本为11.0.2,我安装的版本为11.7,解释一下为什么,因为cuda11.0.2对应的pytorch最高版本为1.7,而yolov5算法在跑过程中要求版本是>=1.8.0,所以cuda11.0.2不满足要求。

(2)Anaconda安装省略请参照这个博主Anaconda安装部分

史上最详细yolov5环境配置搭建+配置所需文件_yolov5环境搭建-CSDN博客

安装好后如下所示打开Anaconda powershell Prompt在这个命令端下创建虚拟环境

3安装pytorch

(1)创建虚拟环境

这里的new1为虚拟环境名称,虚拟环境名字随意去,python版本要求3.8

conda create -n new1 python=3.8

(2)激活虚拟环境new1

conda activate new1

(3)安装pytorch

在查阅资料的过程中查到cuda11.7对应的pytorch稳定版本是1.13.1,q请打开下面的网址

download.pytorch.org/whl/torch_stable.html

打开如下所示我下载的版本是cu117/torch-1.13.1%2Bcu117-cp38-cp38-win_amd64.whl

解释一下为什么,因为最开始查询到我的电脑cuda版本是11.7,cu117代表cuda版本,torch-1.13.1%2Bcu117,代表cuda11.7下载的pytorch版本是1.13.1,cp38-cp38代表python版本是3.8,根据cuda版本选择对应的pytorch版本。

 pip install C:\Users\PC\Downloads\torch-1.13.1+cu117-cp38-cp38-win_amd64.whl

4.检验pytorch安装好了没有

(1)打开pycharm,将环境切换为new1

点设置--添加解释器--conda环境。。。按照如图设置虚拟环境,这样在Anaconda powershell Prompt中设置的虚拟环境new1就对应到了pycharm中

 在设置好后就显示了下图,界面右下角显示new,这里我没有生成new1,仍旧使用之前设置的虚拟new环境,要使用new1就点上一步的new1就好。

 同时打开pycharm终端,命令行开始显示new

上述使用创建虚拟环境的命令也可以在这里进行,这里和 Anaconda powershell Prompt是等价的

仍使用以下命令下载pytorch

 pip install C:\Users\PC\Downloads\torch-1.13.1+cu117-cp38-cp38-win_amd64.whl

 验证pytorch安装好了没有输入以下命令显示true,则说明GPU搭建完成

 import torch
 torch.cuda.is_available()

 

4.安装 torchvision,torchaudio

安装torch==1.13.1+cu117 兼容的 torchvision 版本

最初报错!!

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. torchvision 0.19.1 requires torch==2.4.1, but you have torch 1.13.1+cu117 which is incompatible.

这个错误表明你的环境中已经安装了一个与 torch==1.13.1+cu117 不兼容的 torchvision 版本(0.19.1),它依赖于 torch==2.4.1。要解决这个问题,你需要卸载现有的 torchvision 包,然后安装与 torch 1.13.1+cu117 兼容的版本。

以下是如何操作的步骤:

  1. 卸载现有的 torchvision 包:

shell

pip uninstall torchvision
  1. 重新安装与 torch 1.13.1+cu117 兼容的 torchvision 版本。根据你之前的尝试,这个版本应该是 0.14.1+cu117

shell

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-inde

 

 这个显示就是下载好了,这里要注意torch, torchvision,torchaudio要相互兼容

5.GPU跑yolov5算法

权重文件为yolov5n6.pt,github下载即可

python .\detect.py --source .\data\images --weights .\weight\yolov5n6.pt

这个就跑好了,保存在 Results saved to runs\detect\exp15

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2211658.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

Mac 备忘录妙用

之前使用 Windows 的过程中,最痛苦的事是没有一款可以满足我快速进行记录的应用 基本都得先打开该笔记软件,然后创建新笔记,最后才能输入,这么多步骤太麻烦了 在切换到 MacOS 之后,让我惊喜的就是自带的备忘录&#…

【java面经thinking】一

目录 类加载过程 加载: 连接 初始化 GC回收机制(垃圾回收) 区域 判断对象是否存活 回收机制 HashMap 类加载器 加载标识 加载机制 缓存 自定义加载器: JVM内存结构 常量池 string设置成final 按下网址发生 类加…

C语言有关结构体的知识(后有通讯录的实现)

一、结构体的声明 1.1 结构体的定义 结构体是一些值的集合,这些值被称为成员变量。结构的每个成员可以是不同的类型 1.2 结构体的声明 这里以描述一个学生为例: struct stu {char name[10];//名字int age;//年龄char id[20];//学号char sex[5];//性别 }…

TIM定时器(标准库)

目录 一. 前言 二. 定时器的框图 三. 定时中断的基本结构 四. TIM定时器相关代码 五. 最终现象展示 一. 前言 什么是定时器? 定时器可以对输入的时钟进行计数,并在计数值达到设定值时触发中断。 TIM定时器不仅具备基本的定时中断功能,而且…

【LeetCode】708. 循环有序列表的插入

目录 一、题目二、解法完整代码 一、题目 给定循环单调非递减列表中的一个点,写一个函数向这个列表中插入一个新元素 insertVal ,使这个列表仍然是循环非降序的。 给定的可以是这个列表中任意一个顶点的指针,并不一定是这个列表中最小元素的…

2024免费mac苹果电脑清理垃圾软件CleanMyMac X4.15.8

对于苹果电脑用户来说,设备上积累的垃圾文件可能会导致存储空间变得紧张,影响电脑的性能和使用体验。尤其是那些经常下载和安装新应用、编辑视频或处理大量照片的用户,更容易感受到存储空间的压力。面对这种情况,寻找一种有效的苹…

springboot3使用Excel导入数据库数据

一、导入依赖 <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml --><dependency><groupId>org.apache.poi</groupId><artifactId>poi-ooxml</artifactId><version>5.3.0</version></dependency> 二、…

Xilinx远程固件升级(二)——STARTUPE2原语的使用

通过&#xff08;一&#xff09;可以看出&#xff0c;对于远程固件升级实际上是通过调用flash不同区域的bit实现&#xff0c;通过golden image和update image共同保障了系统的稳定性。在项目中如果将flash的时钟直接绑定FPGA后进行约束&#xff0c;在综合编译时是无法通过的。这…

Spark:DataFrame介绍及使用

1. DataFrame详解 DataFrame是基于RDD进行封装的结构化数据类型&#xff0c;增加了schema元数据&#xff0c;最终DataFrame类型在计算时&#xff0c;还是转为rdd计算。DataFrame的结构化数据有Row&#xff08;行数据&#xff09;和schema元数据构成。 Row 类型 表示一行数据 …

MySQL 8.4修改初始化后的默认密码

MySQL 8.4修改初始化后的默认密码 &#xff08;1&#xff09;初始化mysql&#xff1a; mysqld --initialize --console &#xff08;2&#xff09;之后,mysql会生成一个默认复杂的密码&#xff0c;如果打算修改这个密码&#xff0c;可以先用旧密码登录&#xff1a; mysql -u…

Redis set类型 zset类型

set类型 类型介绍 集合类型也是保存多个字符串类型的元素的&#xff0c;但和列表类型不同的是&#xff0c;集合中 1&#xff09;元素之间是⽆序 的 2&#xff09;元素不允许重复 ⼀个集合中最多可以存储 个元素。Redis 除了⽀持 集合内的增删查改操作&#xff0c;同时还⽀持多…

【图书推荐】《R语言医学数据分析实践》

本书重点 梅俏、卢龙、丁健、张晟、黄龙、胡志坚、张琼瑶、林志刚等业内专家联袂推荐。 以公共医学数据为例&#xff0c;精选大量的实用案例&#xff0c;深入浅出地介绍统计建模分析方法。 帮助读者解决医学数据分析中遇到的实际问题。 通过实际操作引导读者入门科研论文数…

生信分析流程:从数据准备到结果解释的完整指南

介绍 生物信息学&#xff08;生信&#xff09;分析是一个复杂的过程&#xff0c;涉及从数据准备到结果解释的多个步骤。随着高通量测序技术的发展和生物数据的迅猛增长&#xff0c;了解和掌握生信分析的标准流程变得尤为重要。这不仅有助于提高分析的准确性&#xff0c;还能优…

HarmonyOS NEXT 应用开发实战(五、页面的生命周期及使用介绍)

HarmonyOS NEXT是华为推出的最新操作系统&#xff0c;arkUI是其提供的用户界面框架。arkUI的页面生命周期管理对于开发者来说非常重要&#xff0c;因为它涉及到页面的创建、显示、隐藏、销毁等各个阶段。以下是arkUI页面生命周期的介绍及使用举例。 页面的生命周期的作用 页面…

7-I2C与AHT20温湿度传感器

I2C与AHT20温湿度传感器 嵌入式领域另一种常见的通信IIC通信&#xff0c;并用其与AHT20传感器进行交互&#xff0c;获取房间的温度与湿度。 I2C有一条用于传递数据的数据线称为SDA&#xff08;Serial Data&#xff09;&#xff0c;另一条是用于提供同步时钟脉冲的时钟线SCL&am…

看图识微分与导数概念。

可建立如草图所示的局部坐标系。增量Δydy余项是草图中曲线的方程&#xff0c;微分dyydx&#xff08;是关于dx的一次函数&#xff09;是草图中切线的方程。草图形象直观地显示曲线Δy不切线dy。

安全可靠测评结果公告(2024年第1号)

大家可以选择对应的数据库&#xff0c;中央处理器&#xff0c;供参考&#xff1b;尤其是 水资源安可系统 智慧农业安可系统 智慧水利安可系统、智慧水务安可系统&#xff0c;企业安可系统 等参考使用

# 在执行 rpm 卸载软件使用 nodeps 参数时,报错 error: package nodeps is not installed 分析

在执行 rpm 卸载软件使用 nodeps 参数时&#xff0c;报错 error: package nodeps is not installed 分析 一、问题描述&#xff1a; 在执行 rpm 卸载软件使用 nodeps 参数时&#xff0c;报错 error: package nodeps is not installed 如下图&#xff1a; 二、报错分析&…

Java项目分层思路

Java项目分层思路 一、前言二、了解常见的术语1. 应用开发中使用的术语2. 建模和架构设计层面术语总结 三、如何划分1. 单个module2. 多个module 一、前言 每个人、每个开发团队的规范习惯都不太一样&#xff0c;没有固定标准&#xff0c;合适的才是最好的。 二、了解常见的术…

Python Django 查询集的延迟加载特性

Django 查询集的延迟加载特性 一、引言 在 Django 的开发过程中&#xff0c;查询集&#xff08;QuerySet&#xff09;是我们与数据库进行交互的重要工具。查询集提供了一种高效的方式来检索和操作数据库中的数据&#xff0c;且能够进行懒加载&#xff08;Lazy Loading&#x…