ubuntu24.04+5070ti训练yolo模型(2)

三、cuda安装

sudo apt install nvidia-cuda-toolkit

也可通过官网下载cuda和cudnn最新版本手动安装

安装后查看

nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2023 NVIDIA Corporation

Built on Fri_Jan__6_16:45:21_PST_2023

Cuda compilation tools, release 12.0, V12.0.140

Build cuda_12.0.r12.0/compiler.32267302_0

四、yolo模型训练

下载模型https://github.com/ultralytics/ultralytics

pip install ultralytics

出现error: externally-managed-environment

这个错误信息表明你正在尝试在一个由Ubuntu系统或某些系统级服务管理的Python环境中使用pip来安装包,而这个环境是"外部管理"的,意味着它可能已经被Ubuntu的某些部分或其他系统级服务所依赖。直接在这个环境中使用pip安装或更新包可能会破坏这些系统级服务或应用的运行。

sudo apt install python3-venv

建立虚拟环境yolo:python3 -m venv yolo

激活:source yolo/bin/activate

再次安装正常pip install ultralytics

训练参照说明即可。

Build a new model from YAML and start training from scratch

yolo detect train data=coco8.yaml model=yolo11n.yaml epochs=100 imgsz=640

Start training from a pretrained *.pt model

yolo detect train data=coco8.yaml model=yolo11n.pt epochs=100 imgsz=640

Build a new model from YAML, transfer pretrained weights to it and start training

yolo detect train data=coco8.yaml model=yolo11n.yaml pretrained=yolo11n.pt epochs=100 imgsz=640

在训练yolo11x模型时 torch.OutOfMemoryError: CUDA out of memory.

查看batch默认值为16,改为12正常

yolo detect train data=yoloDjW.yaml model=yolo11x.yaml epochs=100 imgsz=640 batch=12

相关推荐
小龙报1 分钟前
《赋能AI解锁Coze智能体搭建核心技能(1)--- 初识coze》
人工智能·语言模型·数据分析·交互·文心一言·机器翻译·coze
love530love15 分钟前
【笔记】Podman Desktop 部署 开源数字人 HeyGem.ai
人工智能·windows·笔记·python·容器·开源·podman
CoookeCola23 分钟前
开源图像与视频过曝检测工具:HSV色彩空间分析与时序平滑处理技术详解
人工智能·深度学习·算法·目标检测·计算机视觉·开源·音视频
董厂长27 分钟前
综述:deepSeek-OCR,paddle-OCR,VLM
人工智能·计算机视觉
禁默35 分钟前
基于金仓KFS工具,破解多数据并存,浙人医改造实战医疗信创
数据库·人工智能·金仓数据库
云卓SKYDROID41 分钟前
无人机动力学模块技术要点与难点
人工智能·无人机·材质·高科技·云卓科技
胡萝卜3.043 分钟前
C++面向对象继承全面解析:不能被继承的类、多继承、菱形虚拟继承与设计模式实践
开发语言·c++·人工智能·stl·继承·菱形继承·组合vs继承
Danileaf_Guo1 小时前
29瓦功耗运行140亿参数模型!Mac mini M4的AI能效革命
人工智能
小草cys1 小时前
华为910B服务器(搭载昇腾Ascend 910B AI 芯片的AI服务器查看服务器终端信息
服务器·人工智能·华为·昇腾·910b
高洁011 小时前
大模型-模型压缩:量化、剪枝、蒸馏、二值化 (4)
人工智能·python·深度学习·aigc·transformer