当前位置:   article > 正文

深度学习目标检测:yolov5环境配置,适合0基础小白,超详细_根据硬件安装yolo环境

根据硬件安装yolo环境

目录

1. 查看电脑状况

2. 安装所需软件

2.1 Anaconda3安装

2.2 Pycharm安装

3. 创建虚拟环境,安装pytorch

3.1 创建虚拟环境

3.2 安装cuda,pytorch

3.2.1 安装cuda

3.2.2 安装pytorch(GPU版)

3.3 安装CPU版本

4. 配置yolov5环境

4.1 下载yolov5源码

4.2 安装yolov5所需库

4.3 pycharm打开yolov5

5.  验证环境


1. 查看电脑状况

深度学习的训练对于电脑显卡要求较高,若电脑没有独立显卡(N卡)或者是AMD的显卡无法使用GPU进行训练,只可以用CPU进行训练,会慢很多。若不清楚电脑有无显卡可以打开任务管理器,点击性能往下找GPU,就可以看到是否有独立显卡,不清楚是否为独立显卡则搜索型号。

若有独立显卡则安装GPU版本的pytorch,若无独立显卡则安装CPU版本的pytorch。

2. 安装所需软件

推荐安装Anaconda3+Pycharm,会安装或者安装过了则跳过这一步

2.1 Anaconda3安装

Anaconda3由于是国外网站下载较慢,推荐通过清华镜像源安装

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source MirrorIndex of /anaconda/archive/ | 清华大学开源软件镜像站,致力于为国内和校内用户提供高质量的开源软件镜像、Linux 镜像源服务,帮助用户更方便地获取开源软件。本镜像站由清华大学 TUNA 协会负责运行维护。https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

选择带有Anaconda3...Windows...exe 字样的进行下载,建议从最新版往后边找几个,最新版一般不够稳定。下载完之后打开Anaconda3进行安装,一直点下一步,选Just Me,安装路径可以直接复制粘贴修改到  D:\Anaconda3  ,也可以修改到其他路径。

2.2 Pycharm安装

安装Pycharm可以直接去官网下载,速度较快,

下载 PyCharm:JetBrains 出品的用于数据科学和 Web 开发的 Python IDE现在最新版本的 PyCharm,适用于 Windows、macOS 或 Linux。https://www.jetbrains.com/zh-cn/pycharm/download/?section=windows

往下拉下载第二个免费版就可以。下载完之后安装,一直点下一步,遇到选择路径 修改路径到D盘或者其它除C盘,再遇到勾选的选项全部勾选就可以。

3. 创建虚拟环境,安装pytorch

3.1 创建虚拟环境

按下Win键,输入anaconda prompt,打开推荐的应用

默认进入的是base环境,此时需要新建环境,在新建环境之前建议更改默认的pip源和conda源。

更改代码如下,直接输入即可,这里选择更换的是中科大源

  1. conda config --remove-key channels
  2. conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  3. conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  4. conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
  5. conda config --set show_channel_urls yes
  6. pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

此时新建虚拟环境(需要关闭加速软件),这里创建一个名为yolov5,python版本为3.9的虚拟环境,也可以修改为其他名或者python版本

conda create -n yolov5 python=3.9

注意:此时如果报错

UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel anaconda/pkgs/free <https://mirrors.ustc.edu.cn/anaconda/pkgs/free>

The channel is not accessible or is invalid.

You will need to adjust your conda configuration to proceed.
Use `conda config --show channels` to view your configuration's current state,
and use `conda config --show-sources` to view config file locations.

等错误,需要修改.condarc文件的内容,具体路径为:c:\users\用户名\.condarc(找不到请显示隐藏文件)

使用记事本打开,复制下面这一段进去替换掉原本所有的内容,替换完之后保存即可

  1. channels:
  2. - defaults
  3. show_channel_urls: true
  4. channel_alias: http://mirrors.tuna.tsinghua.edu.cn/anaconda
  5. default_channels:
  6. - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  7. - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  8. - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  9. - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  10. - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
  11. custom_channels:
  12. conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  13. msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  14. bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  15. menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  16. pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  17. simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

此时再次创建yolov5虚拟环境

conda create -n yolov5 python=3.9

显示如下则正常创建,输入y

创建完之后输入

conda activate yolov5

进入yolov5环境之中,此时有NVIDIA显卡安装GPU版看3.2,其它需要安装CPU版看3.3。

3.2 安装cuda,pytorch

3.2.1 安装cuda

在安装pytorch前需要安装cuda,(若无显卡则跳过)下载cuda前需要先查看显卡支持的CUDA版本最高是多少,按下win+r键,输入cmd,在打开的页面输入:nvidia-smi ,即可查看

上图红框位置显示即为cuda最高支持版本,本教程最高版本达到11.8即可,若没有达到则需要更新显卡驱动。

CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive 在这个网站挑选下载或者直接点击下面链接下载

https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_522.06_windows.exehttp://xn--ghqu9gt28agj1a

下载完之后运行安装,建议安装到默认路径,所以C盘需要留有20G以上的存储空间,一直点击下一步,直到出现这个界面点击自定义,然后全部勾选再一直下一步即可。

安装完成后可以再次在cmd里输入命令:nvcc -V 查看,如下显示即安装成功

有安装cudnn的需求可以自行百度替换文件安装,只用几次有没有区别不大

3.2.2 安装pytorch(GPU版)

打开prompt 输入conda activate yolov5进入yolov5环境,之后输入即可安装pytorch,耐心等待安装完成(需要注意关闭加速软件否则会下载失败)

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

3.3 安装CPU版本

有显卡的按照上述流程安装后即可跳过本部分,无显卡则需要按照下述方法进行,

打开prompt ,进入yolov8环境之后,输入如下命令即可安装cpu版本的pytorch

conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装完成则环境配置完成

4. 配置yolov5环境

4.1 下载yolov5源码

yolov5源码地址:

https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/709342

推荐阅读
相关标签
  

闽ICP备14008679号