技术专栏
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 最新版anaconda移除了activate命令!!!
source 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 -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
再依次安装:
pip install tqdm
pip install scipy
pip install pyyaml
pip install matplotlib
pip install opencv-python==4.1.2.30
pip install requests
pip install seaborn
pip 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环境下输入:
python detect.py
会默认下载最小的yolov5s.pt文件
检测结果如下:
总结
总的来说还是比较简单的,除了yolov5s.pt模型,yolov5还有
声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。
红包
点赞
收藏
评论
打赏
- 分享
- 举报
评论
0个
手气红包
![易百纳技术社区](https://ebaina.oss-cn-hangzhou.aliyuncs.com/production/direct/resource/202306/14/empty.png)
相关专栏
-
2024-03-05 15:21:43
-
浏览量:494次2023-12-29 14:41:17
-
浏览量:854次2024-01-05 16:46:11
-
浏览量:616次2024-01-10 15:17:17
-
浏览量:1073次2023-07-20 11:05:58
-
浏览量:1732次2017-12-25 00:00:33
-
浏览量:667次2024-03-05 15:05:36
-
浏览量:2412次2018-10-01 12:59:37
-
浏览量:3274次2022-03-19 10:58:55
-
浏览量:599次2024-03-06 16:15:59
-
浏览量:663次2023-12-19 16:06:28
-
浏览量:1238次2024-01-05 17:00:10
-
2024-02-02 14:41:10
-
浏览量:7480次2022-07-29 18:09:52
-
浏览量:752次2024-02-19 15:26:47
-
浏览量:594次2023-12-27 15:46:55
-
浏览量:399次2023-06-03 16:03:04
-
浏览量:742次2024-02-20 13:54:36
-
浏览量:557次2023-07-05 10:12:20
置顶时间设置
结束时间
删除原因
-
广告/SPAM
-
恶意灌水
-
违规内容
-
文不对题
-
重复发帖
打赏作者
![易百纳技术社区](https://ebaina.oss-cn-hangzhou.aliyuncs.com/res/images/user/avatar/202310/13/2wC0ygD9r3PcUhKqksb3Su8poojY93Eync2T8FMT.png)
保持微笑
您的支持将鼓励我继续创作!
打赏金额:
¥1
![易百纳技术社区](/img/web/choose_active.png)
¥5
![易百纳技术社区](/img/web/choose_active.png)
¥10
![易百纳技术社区](/img/web/choose_active.png)
¥50
![易百纳技术社区](/img/web/choose_active.png)
¥100
![易百纳技术社区](/img/web/choose_active.png)
支付方式:
![易百纳技术社区](/img/web/paywx.png)
举报反馈
举报类型
- 内容涉黄/赌/毒
- 内容侵权/抄袭
- 政治相关
- 涉嫌广告
- 侮辱谩骂
- 其他
详细说明
审核成功
发布时间设置
发布时间:
请选择发布时间设置
是否关联周任务-专栏模块
审核失败
失败原因
请选择失败原因
备注
请输入备注