这篇教程yolov5 win10 CPU与GPU环境搭建过程写得很实用,希望能帮到您。
前言最近实习任务为黑烟检测,想起了可以尝试用yolov5来跑下,之前一直都是用的RCNN系列,这次就试试yolo系列。
一、安装pytorch
1.创建新的环境打开Anaconda Prompt 命令行输入 创建一个新环境,并激活进入环境。 # 创建了名叫yolov5的,python版本为3.8的新环境conda create -n yolov5 python=3.8# 激活名叫yolov5的环境conda activate yolov5
2.下载YOLOv5 github项目下载地址为: https://github.com/ultralytics/yolov5 如果安装了git 可以使用git clone https://github.com/ultralytics/yolov5 ,没有的话直接下载zip 压缩包也行,把压缩包解压到指定目录就行。
3.安装相关依赖库和包查看一下requirements.txt 里面的内容并下载所有的依赖包 依次安装,建议可以用清华源进行安装: 先把清华源设置成默认: pip install pip -Upip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple 再依次安装: pip install tqdmpip install scipypip install pyyamlpip install matplotlibpip install opencv-python==4.1.2.30pip install requestspip install seabornpip install pandas 安装pytorch需要注意一下: 还是要先换源: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 然后再用下面的命令代码确认 conda config --set show_channel_urls yes 之后进入官网https://pytorch.org/ 找到合适的版本,如果你是安装cpu版
 运行conda install pytorch torchvision torchaudio cpuonly 如果你是安装GPU版
 运行conda install pytorch torchvision torchaudio cudatoolkit=10.1 这里个人的计算机配置不一样,就拿我的来说cuda10.1 可以运行,cuda11 无法运行GPU版,这里可以多试几次找到适合自己的版本。
4.验证在刚刚建好的yolov5 环境下启动python python#输入库import torch#查看版本print(torch.__version__) CPU 版如果到这步不报错,就说明安装成功了。
GPU 版需要再运行一些代码查看
#查看gpu是否可用torch.cuda.is_available()#返回设备gpu个数torch.cuda.device_count() 一切正常的话,GPU版的pytorch就安装成功了
二、运行detect.py文件在建好的yolov5 环境下输入: 会默认下载最小的yolov5s.pt 文件 检测结果如下:


总结总的来说还是比较简单的,除了yolov5s.pt 模型,yolov5还有 
V3.1权重文件下载不下来的,可以通过下面的链接下载 http://xiazai.jb51.net/202104/yuanma/yolov_jb51.rar
到此这篇关于yolov5 win10 CPU与GPU环境搭建过程的文章就介绍到这了,更多相关yolov5 win10 CPU与GPU环境搭建内容请搜索51zixue.net以前的文章或继续浏览下面的相关文章希望大家以后多多支持51zixue.net! win10系统配置GPU版本Pytorch的详细教程 win10+anaconda安装yolov5的方法及问题解决方案 |