YOLOv5安装以及环境配置

目录

1YOLOv5汉化版代码下载地址:https://github.com/wudashuo/yolov5

2环境搭建

3安装步骤

4下载yolov5代码


1YOLOv5汉化版代码下载地址:https://github.com/wudashuo/yolov5

2环境搭建

1.支持cuda的Nvidia显卡(显存越大越好)

2.python 3.8以上(推荐anaconda)

3.Cuda 10.2

4.pytorch

5pycharm

6.YOLO v5源码

3安装步骤

3.1python安装

进入python官网地址:https://www.python.org 选择需要的版本下载,最新版本3.9,也可以下载3.8或3.7的版本

 3.2anaconda安装

进入anaconda官网 https://www.anaconda.com/products/individual#Downloads 下载对应版本,一般windows选择64-Bit。

这里下载好anaconda之后自己创建一个环境来运行yolov5的程序 ,这里小编创建的为torch17环境.

3.3Cuda 10.2

下载地址:https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

这里不要想着下载最新版本版本的cuda,因为最新版本的可能不支持对应版本的pytorch,导致不能使用显卡。

 安装cudnn,这里安装cudnn需要注册并且登录才能下载,进去之后选择对应cuda10.2版本的即可。具体安装过程看下面这个博客。

参考这篇博客,写的非常详细,只需要按照自己需要安装的版本即可,博客地址:https://blog.csdn.net/sinat_23619409/article/details/84202651

3.4Pytorch安装

官网链接https://pytorch.org/get-started/locally/

 使用windows+R打开命令终端,然后输入activate torch17 ,我们可以看到已经打开创建好的环境(torch17).

然后辅助黏贴到命令行

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装pytorch。

如果使用conda 安装不了torch的话,这里需要手动下载并安装安装torch,根据对应版本的pytorch安装,各个版本的torch下载地址:

https://download.pytorch.org/whl/torch_stable.html

3.5pychram下载

下载地址:https://www.jetbrains.com/pycharm/download/#section=windows

选择community版本的及社区版。

4下载yolov5代码

在guithub上下载yolov5代码之后,解压,用pychram打开文件夹,在pycharm中选择自己创建的环境

然后安装requirements.txt里面需要的库,可以一起安装,也可以分布安装,如果在某个安装环节出现问题,直接百度该问题即可。

 

以上就是yolov5配置的全部过程,如果对您有用,请给个赞,遇到什么问题,欢迎评论。 

  • 3
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是 yolov5gpu 版本环境配置的步骤: 1. 安装 Anaconda 首先,需要安装 Anaconda。在 Anaconda 官网下载对应版本的安装包,然后按照指示安装。 2. 创建虚拟环境 打开 Anaconda Prompt 终端,输入以下命令创建一个名为 yolov5gpu 的虚拟环境: ``` conda create --name yolov5gpu python=3.8 ``` 3. 激活虚拟环境 创建完虚拟环境后,需要激活这个环境。在 Anaconda Prompt 终端中输入以下命令: ``` conda activate yolov5gpu ``` 4. 安装 PyTorch yolov5gpu 需要使用 PyTorch 神经网络框架。可以通过以下命令安装 PyTorch: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia ``` 其中,cudatoolkit=11.1 表示使用 CUDA 11.1 版本,可以根据自己的显卡情况选择对应的版本。 5. 安装其他依赖库 yolov5gpu 还需要安装其他依赖库,可以通过以下命令安装: ``` pip install opencv-python matplotlib pillow PyYAML ``` 6. 下载 yolov5 代码 可以从 yolov5 的 GitHub 仓库下载 yolov5 代码。在 Anaconda Prompt 终端中输入以下命令: ``` git clone https://github.com/ultralytics/yolov5.git ``` 7. 测试 完成以上步骤后,可以通过以下命令测试 yolov5 的运行情况: ``` cd yolov5 python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images/bus.jpg ``` 其中,--weights 参数指定 yolov5 模型的权重文件,--img 参数指定输入图像的大小,--conf 参数指定置信度阈值,--source 参数指定输入图像的路径。 以上就是 yolov5gpu 版本环境配置的步骤。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值