YOLOv5环境配置

0 安装Anaconda和pycharm

略…

1 创建YOLOv5虚拟环境

  1. 找到安装好的Anaconda,打开Anaconda Prompt

image-20230803101924719

  1. 在命令面板中输入一下代码
conda create -n yolov5 python=3.8
  1. 输入y

image-20230803102327811

  1. 出现下图表示创建完成

image-20230803102701943

  1. 输入activate yolov5激活环境(该窗口不要关闭)

image-20230803102913344

2 安装pytorch

2.1 查看电脑支持CUDA版本

  1. 电脑桌面右键,打开NVIDIA控制面板

image-20230803104408122

  1. 在【帮助】菜单拦下,打开【系统信息】

image-20230803104654479

  1. 在【组件】窗口,查看自己的CUDA支持版本

我这里最高支持11.4版本

image-20230803104745264

2.2 下载pytorch

  1. 进入PyTorch官网,选择适合自己的版本

需要说明一下,30系列显卡,必须安装11以上的版本,10版本不再支持

image-20230803105038763

  1. 这里我选择安装11.3版本

image-20230803105338301

  1. 在开始之前,先将pip下载切换为国内镜像(在 pip install 后面添加 -i https://pypi.tuna.tsinghua.edu.cn/simple)

这里选择清华源,临时镜像,如果需要换成永久的,可以进清华源的官网根据提示进行切换

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

image-20230803113738501

  1. 内容有点大,耐心等待安装完成

出现successfully表示安装成功

image-20230803114212772

3 下载yolov5源码

GitHub地址:YOLOv5

国内GitHub访问慢,可以使用Watt Toolkit软件进行加速(这个不是梯子,免费的,加速功能强大)

  1. 进入GitHub后,在右侧找到releases,进入下载zip(下载的v7)

image-20230803120019909

image-20230803120109208

  1. 找一个文件夹,将下载的压缩包解压出来

image-20230803120318830

4 PyCharm部署运行

  1. 用PyCharm打开刚才解压出来的文件夹

image-20230803122534135

  1. 先不创建虚拟环境,进入后手动添加环境

image-20230803122501998

image-20230803171522945

image-20230803171544230

image-20230803171746039

  1. 打开requirements.txt文件,将下图所示的部分进行修改

image-20230803122939899

  1. 在 PyCharm打开终端,输入activate yolov5激活环境

image-20230803123134358

image-20230803123159927

  1. 安装依赖环境
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  -r requirements.txt 

安装过程中,会出现一堆warring,这个不用管

出现下图所示内容代表安装成功

image-20230803123654906

  1. 运行测试
python detect.py

到下图这一步,下载较慢,这时可以将下载地址复制下来,在浏览器上下载

image-20230803123957619

输入Ctrl+C,结束命令行下载任务

image-20230803124314690

将刚才下载的文件放到文件夹下,然后在pycharm命令行中重新执行python detect.py

image-20230803124412661

出现下图信息,表示环境搭建成功

image-20230803124535603

打开runs文件夹夹,监测效果如图所示

python detect.py

image-20230803124412661

出现下图信息,表示环境搭建成功

image-20230803124535603

打开runs文件夹夹,监测效果如图所示

image-20230803124711601

猜你喜欢

转载自blog.csdn.net/qq_61228493/article/details/132207846