yoloV5实战笔记—环境搭建(一)

时间:2024-11-08 19:41:54

一、安装miniconda

从清华源进行下载
https://mirrors.tuna.tsinghua.edu.cn/
具体命令参考,注意修改pip国内镜像地址
https://mirrors.tuna.tsinghua.edu.cn/help/AOSP/
创建环境,指定python版本

conda create -n demo python=3.9

激活环境

conda activate demo

即可激活demo项目。以下步骤也都是在demo环境下执行。

二、安装pytorch,选择跟自己电脑匹配的版本

pytorch和torchvision下载地址

https://download.pytorch.org/whl/torch/
https://download.pytorch.org/whl/torchvision/

官网 https://pytorch.org/
可以提前用迅雷下载好pytorch,国内下载确实比较慢
使用本地安装

pip install torch d:/test/torch-2.5.1+cu118-cp39-cp39-win_amd64.whl

三、下载yolov5

https://github.com/ultralytics/yolov5
进入到项目目录,安装必须的依赖
注意参考pytorch和vision的版本,在requirements.txt文件中直接配置死。
在这里插入图片描述
但是最后还是会有问题,如果安装的pytorch是cuda版本的,那么安装的torchvision也必须是cuda版本。torchvision地址为: https://github.com/pytorch/vision

pip install -r requirements.txt

四、安装cuda版torchvision

pip install D:\test\torchvision-0.20.0+cu118-cp39-cp39-win_amd64.whl

五、测试运行

默认会去下载yolov5s.pt模型文件,也可以提前下载下来,放到项目目录下面,再执行即可。
在这里插入图片描述
下载完成之后,在项目目录中的结构为:
在这里插入图片描述
执行命令

python detect.py

正常情况,即可看到输出,并且在runs\detect文件夹也可以看到输出的图片。
在这里插入图片描述

六、错误排查

1、torchvision报错

报错:NotImplementedError: Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend. This could be because the operator doesn’t exist for this backend
torch、torchvision、torchaudio都有GPU版本,有时报错可能是装的CPU版本,最好按官网命令统一装,重新装torchvision即解决。