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

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.改完这三个运行成功后记得改回去,否则影响后续重新训练模型

相关推荐
长弓三石几秒前
企业级 Harness 平台的 IM 接入:从选型到上线,一篇讲完
人工智能
冬奇Lab6 分钟前
开源项目第243期:ArtCraft —— 给艺术家的 AI 创作 IDE,先搭场景再生成,把「抽卡」变成「手艺」
人工智能·开源
闽林致远6 分钟前
从文案语义到素材自动获取的工程链路-AutoVideoHub
人工智能
kkkkkkkkkk_Z7 分钟前
新手自学嵌入式 | 学习日记:Linux驱动复习之系统移植(U-Boot篇)
linux·运维·学习
Rabitebla10 分钟前
【系统性学习笔记】Linux 权限篇(二):文件属性、权限修改与目录权限的底层逻辑
linux·c++·笔记·学习·算法
摩羯123011 分钟前
AI提示词备查与原始输入输出记录:生成式人工智能合规的可验证实现
人工智能·aigc
用户19172912708313 分钟前
无 API 老系统怎么让 Agent 接管:Computer Use 四层架构与可运行代码
人工智能
AICoder码农王21 分钟前
如何写 Skill 第8课,多 skill 体系治理:从治一个到治一片
人工智能
远离UE422 分钟前
半拉格朗日学习笔记 动力学 实时渲染
人工智能·算法
天远数科24 分钟前
Python数据工程:利用天远企业三要素验证优化大宗商品核心企业供应链入驻合规体验
人工智能