前提: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

GitHub 加速计划 / yo / yolov5
49.42 K
16.03 K
下载
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 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐