news 2026/9/3 8:12:10

Inter的电脑使用GPU加速训练

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Inter的电脑使用GPU加速训练

换新电脑就是不一样,但是突然发现是inter的显卡,下面说一下如何使用inter的显卡加速训练。先大致步骤说一下:将ultralytics软件包中的torch_utils.py中的select_device函数修改一下,然后使用训练代码进行进行训练就可以了。

一、环境

现在的YOLO直接下载ultralytics这个软件库,再加上数据集,就可以了。

首先找到下载好的ultralytics软件包中需要修改的函数代码文件,下面是以conda环境为例的路径:

Anaconda3\envs\你的环境名\Lib\site-packages\ultralytics\utils\torch_utils.py

找到后用编译器打开,在183行找到如图的代码加上一个else语句让他可以识别到xpu就可以了,

代码如下:

elif device == 'xpu' or device.startswith('xpu:'): # 检查 Intel XPU 是否可用 if hasattr(torch, 'xpu') and torch.xpu.is_available(): # 解析设备索引,例如 ‘xpu:0’ device_idx = 0 if ':' in device: device_idx = int(device.split(':')[1]) # 检查索引是否有效 if device_idx < torch.xpu.device_count(): device_obj = torch.device(f'xpu:{device_idx}') if verbose: print(f'使用 Intel XPU 设备: xpu:{device_idx} ({torch.xpu.get_device_name(device_idx)})') return device_obj else: raise ValueError(f"无效的 XPU 设备索引 {device_idx},可用设备数为 {torch.xpu.device_count()}") else: raise ValueError(f"请求使用 Intel XPU (‘device={device}’),但 XPU 不可用。请检查驱动和 IPEX 安装。")

然后就保存退出就可以了。

二、训练代码

这样就可以使用常规训练代码训练了,但是在训练1轮之后会报错,这个 _clear_memory 是一个通用工具函数,它在每个训练周期(epoch)结束后都可能被调用,目的是防止显存(VRAM)峰值。这个会自动调用CUDA,所以在训练代码开头加上一个猴子补丁,下面是训练代码:

# ============ 【1. 必加:猴子补丁,解决所有CUDA硬编码问题】 ============ import os os.environ['NUM_WORKERS'] = '0' # Windows必须 os.environ['PYTHONWARNINGS'] = 'ignore' import torch import ultralytics.engine.trainer as trainer_module # --- 修补 _get_memory 函数 --- _original_get_memory = trainer_module.BaseTrainer._get_memory def _patched_get_memory(self, fraction=False): """永远返回一个极低的内存使用率,让框架跳过清理,避免调用 torch.cuda""" return 0.01 if fraction else 1024 # 返回1%使用率或1024字节 trainer_module.BaseTrainer._get_memory = _patched_get_memory print("✅ 关键补丁已应用:内存检查函数已替换,规避CUDA调用。") # ============ 【2. 正常训练流程】 ============ from ultralytics import YOLO # 设置设备 device = torch.device('xpu:0') print(f"训练设备: {device}") # 加载模型并移动 model = YOLO('yolov8n.pt') # 请确保此权重文件存在 model.model.to(device) print(f"模型已移至: {next(model.model.parameters()).device}") # 开始训练 results = model.train( data='data/data.yaml', # 你的数据集配置文件 epochs=5, imgsz=640, batch=32, device='xpu', # 框架用此参数处理数据位置 workers=0, amp=False, # 必须关闭,避免AMP相关CUDA调用 half=False, # 必须关闭 verbose=True, name='yolo_xpu_final_fixed', plots=True, ) print(f"\n🎉 训练完成!结果保存在: {results.save_dir}")

你可以根据自己的数据集来修改2中的参数(注释的不要修改),然后就可以开始在新电脑上进行GPU加速训练了。非常的好用,快速解决问题。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/3 0:51:26

本地运行本教程 Fiori Elements 应用时弹出 Sign in 对话框的真正原因

有朋友提问,本地命令行 npm run start 运行启动教程提供的 Fiori Elements 应用之后,看到下面截图的弹出框,是什么意思? 这里只需要输入 ES5 系统的用户名和 credential 即可。 2. 在 ES5 系统注册用户以获得 Fiori Elements 开发教程测试 OData 服务的访问账号 如果大家想…

作者头像 李华
网站建设 2026/9/2 23:45:08

露,AI人工智能自发活动分析系统 AI人工智能自发活动视频分析系统

自发活动视频分析系统主要用于观察和分析实验动物在新环境中的神经精神状态及行为表现。该系统可记录动物在开阔场中的运动轨迹、活动时间、中央区域停留时间等参数&#xff0c;以评估其探索行为和焦虑样情绪。安徽&#xff0c;正华生物&#xff0c;露实验动物面对新环境时&…

作者头像 李华
网站建设 2026/9/2 20:37:27

告别无效联系!天下工厂查询工具真实体验报告

在B2B销售业务、采购工作或者供应链拓展过程中&#xff0c;让人最感到灰心失望的事情&#xff0c;其实并不是没有客户资源&#xff0c;其实事实上&#xff0c;真正让人沮丧的是&#xff0c;你联系了足足一百个号称“厂家”的对象&#xff0c;可结果却发现其中九十家都是中间商&…

作者头像 李华
网站建设 2026/9/2 20:41:08

突发|Meta 官宣引入 Manus:创始人出任副总裁,核心团队整体并入

一条来自硅谷的重磅消息正在 AI 圈迅速扩散。Meta 与 Manus 官方几乎在同一时间发布公告&#xff0c;确认 Manus 正式加入 Meta 体系。随后&#xff0c;Meta 高层也在社交平台同步释放信号&#xff0c;消息真实性已被多方确认。这并非一则普通的合作新闻&#xff0c;而是一场被…

作者头像 李华
网站建设 2026/9/2 21:33:22

2026 网络安全学习指南,详尽全面!建议收藏学习

学习的问题 首先咱们聊聊&#xff0c;学习网络安全方向通常会有哪些问题 1、打基础时间太长 学基础花费很长时间&#xff0c;光语言都有几门&#xff0c;有些人会倒在学习 linux 系统及命令的路上&#xff0c;更多的人会倒在学习语言上&#xff1b; 2、知识点掌握程度不清楚…

作者头像 李华