返回顶部
首页 > 资讯 > 后端开发 > Python >YOLOv7保姆级教程!(个人踩坑无数)----环境配置
  • 152
分享到

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

python人工智能深度学习 2023-09-30 22:09:02 152人浏览 泡泡鱼

Python 官方文档:入门教程 => 点击学习

摘要

      一、前言:         博主本人是做目标检测的,使用YOLO较多,特此写一个详细博客,一起学习一下。 二、环境搭建:         深度学习相关环境的搭建可谓是相当重要的,不仅是深度学习的入门第一关,更是影响你整个模型复现

     

一、前言:

        博主本人是做目标检测的,使用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

--结束END--

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

本文链接: https://lsjlt.com/news/422075.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作