一、安装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即解决。