yolov5

最新推荐文章于 2024-07-22 11:47:36 发布

程序猿007007

最新推荐文章于 2024-07-22 11:47:36 发布

阅读量313

点赞数

分类专栏： yolov5 文章标签：深度学习 pytorch

本文链接：https://blog.csdn.net/weixin_49175350/article/details/117078306

版权

2 篇文章 1 订阅

订阅专栏

安装环境

YOLOv5是在YOLOv3的基础上改的，可以实现目标检测，也可以实现分类。训练时间相较于其他模型大大的减少。

1.安装环境最好使用anaconda建立虚拟环境，anaconda下载地址。
2.下载完成之后进行安装，这里给出其他博主安装anaconda的安装教程。
3.安装完成之后吗，点击Windows的开始键，找到Anaconda Prompt，然后打开。
在这里插入图片描述

2.1创建环境
点击Anaconda Prompt，然后创建一个虚拟环境，并且激活环境。

# 创建了名叫pytorch的，python版本为3.8的新环境
conda create -n pytorch1.7 python=3.8
# 激活名叫pytorch的环境
activate pytorch

在这里插入图片描述

2.2下载pytorch
去pytorch的官网https://pytorch.org/

这里安装是pytorch1.7 搭配的是yolov54.0，pytorch1.6搭配的是yolov53.0，不要安装pytorch1.8，因为一些镜像源没有对应的包。

下拉找到Previous versions of PyTorch，然后点击，去找pytorch1.7.1版本。
在这里插入图片描述

复制上图的命令到Anaconda Prompt，在这一步，需要多等一会儿。

在虚拟环境输入Python，启动Python，导入Python库。

# 启动
python
#输入库
import torch

查看是否可用

#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()

在这里插入图片描述
到这里，pytorch-gpu版本就安装成功了。

关注

专栏目录