学习过程中遇到的 部分问题及解决办法

1.安装build wheel时报错:

The detected CUDA version (12.1) mismatches the version that was used to compile

PyTorch (11.7). Please make sure to use the same CUDA versions.

由于cuda版本和 当前虚拟环境中的pytorch-cudatoolkit版本不同,

解决:

安装好对应低版本cuda 11.7后,在当前命令行输入export CUDA_HOME=/usr/local/cuda-11.7

这样检测cuda版本时用的就是11.7版本,换一个窗口就需要重新export

2.使用yolov10不能resume,即不能从上一次断点继续训练:

解决:在model.py中

1.注释掉

self.model = self.trainer.model

2.在trainer.py中resume_trainning加入ckpt = torch.load('./runs/detect/train15/weights/last.pt')

3.check_resume函数中将原来的#resume = self.args.resume 改为resume='./runs/detect/train15/weights/last.pt'

4.改完这三个运行成功后记得改回去,否则影响后续重新训练模型

相关推荐
品牌测评2 小时前
当AI编码平台走向订阅制与多模型聚合:以Alaya Code为例的技术观察
人工智能
茶马古道的搬运工2 小时前
AI 深度技能之-解读DeepSeek Harness(一)- 初见
人工智能
oort1232 小时前
OortCodex 是奥尔特云 OortCloudSmart 推出的国产化工程化 AI 编码 Agent(奥尔特云编码智能体)
大数据·人工智能·算法
TheBestRucy2 小时前
基于Dify的旅游攻略&王者荣耀攻略智能助手项目
服务器·开发语言·人工智能·python·算法·旅游
你想知道什么?2 小时前
朴素贝叶斯算法-学习笔记
笔记·学习·算法
天吾cc2 小时前
JSON 与 cJSON 学习笔记
笔记·学习·json
chen_zn952 小时前
《WAM 系列》Fast-WAM|视频共训练|跳过未来想象|实时WAM
人工智能·具身智能·vla·世界模型·世界动作模型
能年玲奈喝榴莲牛奶2 小时前
资产和漏洞管理系统(AI)
人工智能·安全·web安全
DS随心转小程序2 小时前
借助 AI 导出鸭简化各类办公场景下文心输出 word 文档全流程操作
人工智能·aigc·word·豆包·deepseek·ai导出鸭
lifallen2 小时前
Orca 与 Emdash:同样管理多个 Agent,差别在谁来调度
人工智能·学习·ai·软件构建·开源软件·ai编程