YOLOv5使用GPU训练
yolov5
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
项目地址:https://gitcode.com/gh_mirrors/yo/yolov5
·
前提:yolov5在CPU训练可以进行后,想改用GPU提速遇到问题。
原因:自己的torch下的是CPU版本。
方法:重新创建一个解释器。(以下一切在命令提示符窗口运行)
step1:conda remove -n yolo --all
step2:conda create -n yolov5 python=3.8
step3: 激活解释器,去PyTorch官网下载对应GPU版本的torch(用pip下载)
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
step4:验证
python
#输入库
import torch
#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
#退出python
quit()
step4:把yolov5文件下的requestments.txt文件里的torch、torchvirsion注释掉
step5:cd到对应位置,按要求安装库
d:
cd D:\WXR\yolov5-master
pip install -r requirements.txt
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
最近提交(Master分支:2 个月前 )
79b7336f
* Update Integrations table
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.zh-CN.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> 26 天前
94a62456
* fix: quad training
* fix: quad training in segmentation 28 天前
更多推荐

所有评论(0)