YOLOv5安装笔记(1)——安装环境(windows版本)
安装环境
YOLOv5是在YOLOv3的基础上改的,可以实现目标检测,也可以实现分类。 训练时间相较于其他模型大大的减少。
1.下载Anaconda
1.安装环境最好使用anaconda建立虚拟环境,anaconda下载地址。
2.下载完成之后进行安装,这里给出其他博主安装anaconda的安装教程。
3.安装完成之后吗,点击Windows的开始键,找到Anaconda Prompt,然后打开。
2.安装pytorch
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,在这一步,需要多等一会儿。
3.验证
在虚拟环境输入Python,启动Python,导入Python库。
# 启动
python
#输入库
import torch
查看是否可用
#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
到这里,pytorch-gpu版本就安装成功了。
4.下载pycharm
下载pycharm开发工具之后对yolov5代码进行运行。点击下载
YOLOv5安装笔记(1)——安装环境(windows版本)
YOLOv5安装笔记(2)——安装其他包和训练coco数据集