Yolov5环境配置

一、安装nVidia驱动和cuda

        nVidia安装驱动官网地址：官方驱动 | NVIDIA

        

        根据自己电脑显卡版本安装nVidia驱动：

        1、右键此电脑选择管理；2、选择设备管理器；3、选择显示适配器查看显卡版本。

nVidia驱动下载完成后进行安装，安装完成后可进行查看；

通过命令窗口可进行查看：win+R,输入cmd打开命令窗口，然后输入以下命令符：

nvidia-smi

``````

右上角CUDA Version显示可支持的最高版本。

二、安装anaconda和配置虚拟环境

1、安装anaconda

前往anaconda官网进行下载软件安装包，官网地址：免费下载 |蟒蛇 (anaconda.com)；这里下载的是最新版本；若需要安装其他版本也可前往镜像进行安装：例如，清华镜像Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror。

安装包下载后双击进入，安装地址建议不要放在C盘当中，懂得都懂；其中最重要的有一步是需要勾选添加Anaconda到PATH环境变量当中；如果没有勾选后续还要自己去编辑系统环境变量中进行编辑。

                               

安装完成之后可在菜单栏下搜索Anaconda，出现于以下框中Anaconda文件则安装成功

                

2、配置虚拟环境

win+R输入cmd，打开命令窗口

若需要换盘符则可输入'D:'     这里以D盘为例

接着输入以下命令激活base环境

activate base

                        

接着输入以下命令创建虚拟环境

conda create -n 'name' python=x.x.x
例如：conda create -n yolo python=3.9

回车后，等候一段时间，当出现（y/n）输入y进行安装

                

当显示下图内容则表示虚拟环境创建完成

                

win+R输入cmd，打开命令窗口，输入以下命令查看自己所创建的虚拟环境

conda env list

`````````

若是要删除所创建的虚拟环境则输入以下命令,然后可再进行查看

conda remove -n yolo --all

                        

三、安装pytorch

首先要激活自己所创建的虚拟环境：输入命令：

conda activate yolo

然后前往pytorch官网下载pytorch软件包：Previous PyTorch Versions | PyTorch，我的cuda版本是10.2，因此我选择的是以下命令，然后等待安装完成即可。

conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 -c pytorch

若不知道自己所安装的cuda版本是多少，可以通过win+R打开命令窗口来进行查看，输入以下命令可查看所安装的版本，然后下载相对应的pytorch。

nvcc -V

                

四、下载yolov5源码以及权重文件

1、yolov5源码下载地址：GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

                

2、权重文件下载

还是当前页面往下拉，找到Releases点进去，进入后点Tags，选择v5.0，进去后往下拉找到Assets，就可以找到所有的权重文件，点击下载即可。

五、安装pycharm以及添加解释器

1、pycharm安装地址：

下载 JetBrains 提供的 PyCharm：用于数据科学和 Web 开发的 Python IDE

安装步骤就不多加赘述了，网上都有许多教程；

2、配置解释器

打开下载好的pycharm软件，新建项目后打开设置，找到python解释器，然后添加本地解释器；

选择conda环境，conda可执行文件选自己所安装的anaconda路径下的conda.exe文件，环境使用你之前所创建的yolo环境，点击确定即可。

六、安装yolov5所依赖的软件包

1、安装依赖包

在pycharm软件中打开终端：

然后根据yolov5源码中的requirements.txt文件，利用pip或者conda指令安装所依赖的软件包：

常用的指令：
pip install matplotlib
pip install numpy
pip install opencv-python
pip install Pillow
pip install PyYAML
pip install scipy
pip install tqdm
pip install tensorboard
pip install seaborn
pip install pandas
pip install thop
pip install pycocotools

2、各类镜像：

豆瓣镜像：

pip install -i http://pypi.douban.com/simple/ some-package
pip install -i http://pypi.douban.com/simple/ some-package --trusted-host pypi.douban.com

其中some-package为安装包的名字，若S需要信任则可使用第二条命令

清华镜像：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host pypi.tuna.tsinghua.edu.cn

阿里镜像：

pip install -i http://mirrors.aliyun.com/pypi/simple/ some-package
pip install -i http://mirrors.aliyun.com/pypi/simple/ some-package --trusted-host mirrors.aliyun.com

中国科技大学镜像：

https://pypi.mirrors.ustc.edu.cn/simple/ some-package

清华镜像：

清华镜像：

七、测试