文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

YOLOv7保姆级教程!(个人踩坑无数)----环境配置

2023-09-30 22:38

关注

     

一、前言:

        博主本人是做目标检测的,使用YOLO较多,特此写一个详细博客,一起学习一下。

二、环境搭建:

        深度学习相关环境的搭建可谓是相当重要的,不仅是深度学习的入门第一关,更是影响你整个模型复现与改进。

2.1 Anaconda:Anaconda主要作为python的虚拟环境,后续也可创建更多不同环境以应对不同需求。

下载官网:Anaconda | The World's Most Popular Data Science Platform

 

点击后根据个人需求,下载适应电脑版本的Anconda。安装Anaconda的过程比较简单,如果有问题可以私信。

2.2 Pycharm:

        pycharm是大多数同学选择编译运行python的平台,比较方便使用。直接点击链接官网下载即可。

下载官网:PyCharm:JetBrains为专业开发者提供的Python IDE

        

直接选择社区版本下载即可, 设置一个路径,正常软件安装即可。

2.3 Cuda

        CUDA在程序中主要是GPU加速使用,如果你只使用CPU对模型训练,此处可以跳过,如果GPU训练同学,请仔细阅读,此处坑较多,当时本人也是踩坑无数。

CUDA官方安装教程:CUDA Installation Guide for Microsoft Windows

CUDnn官方安装教程:Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

桌面右键打开nvidia控制面板

 点击左下角系统信息

 打开组件,查看第三行信息,可以看到你电脑显卡适用的cuda版本

第二种方法:win+r快捷键,输入cmd,打开终端窗口输入nvidia-smi

 此处也可查看cuda版本,最高支持11.4,可以选择11.4以下版本。

CUDA下载地址:CUDA Toolkit Archive | NVIDIA Developer

 

 点击链接选择适合自己电脑的版本,博主本人安装的是11.3版本

 点击进入后,选择系统版本,第一个为本地离线下载,第二个为网络下载。

下载完成后安装:

        注意:第一次为会设置临时解压目录,第二次设置安装目录。

临时解压路径选择默认即可,也可以自定义,安装结束后,临时解压文件夹会自动删除;

 

 

如果是新手小白,那就全都要,如果是安装好几次了,直接第一个。

 

安装到此处后,建议直接截图保存记住安装位置的路径后面需要用。

查看环境变量,点击设置直接搜索查看高级系统设置---》点击环境变量、

就可以看到你的环境变量了,

 

这块就可以看到刚刚安装的11.3的环境变量了。如果没有需要自己添加

这两个变量放在下面这块,方便复制,注意自己的路径即可

NVCUDASAMPLES_ROOT

NVCUDASAMPLES11_0_ROOT

然后查看是否安装成功:

win+r 快捷键输入cmd,输入nvcc --verison或者nvcc -V

 

查看安装cuda的版本号 ,输入set cuda可以查看CUDA设置的环境变量,至此CUDA部分安装完成。

CUDnn下载及安装

官方下载地址:https://developer.nvidia.com/rdp/cudnn-download

注意下载这块需要注册nvidia账号,这个都比较easy,自己操作一下就行了

登录后就能看到不同的CUDnn版本

也可以通过这个链接下载

cuDNN Archive | NVIDIA Developer 

依然是选择适合自己电脑的版本,下载后解压会发现并不是exe文件而是三个文件夹。

 

 接下来比较重要,将这个三个文件夹复制到当时安装cuda的路径中,进行替换。

当时安装的截图中有,或者去系统中的环境变量查看也可以。其实cudnn更像是cuda的一个加速补丁。

然后再次添加路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp

这块的话 根据自己的路径进行修改就可以了。

此时需要验证,win+r 输入cmd打开终端

cd 切换到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe

 

如果看到出现两个pass后,安装cuda与cudnn完成,至此深度学习的最基本环境已经配备,下一篇博客会详细讲解YOLOv7的安装与训练自己的数据集。

来源地址:https://blog.csdn.net/weixin_55749226/article/details/128479694

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯