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

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 分钟前
可见光之外的视界,短波红外成像技术能做什么?
人工智能·计算机视觉·视觉检测
m0_547486665 分钟前
《人工智能导论:深度学习大模型基础》全套PPT课件2026
人工智能·深度学习·大模型
ZhengEnCi8 分钟前
为什么 DeepSeek V4 Flash 可以超过 V4 Pro Preview
人工智能
wordbaby8 分钟前
别问实习生,看回执单——Agent 验证层的核心原则
人工智能
fivebliss9 分钟前
取算存三步细化及延迟指标解析
人工智能·性能优化
铅笔侠_小龙虾11 分钟前
rust 学习(9):结构体
开发语言·学习·rust
武子康12 分钟前
让不同大模型共享一个 Agent:Pi 如何统一 Provider 与 Context Handoff
人工智能
秦先生在广东15 分钟前
`agent-skills`:用工程纪律驯化 AI 编程 Agent 的结构化实践
人工智能
月光船幽幽19 分钟前
门控函数SHS阈值与调制机制解析
人工智能·python·算法
秦先生在广东23 分钟前
Agency-Agents:用结构化「职业操作系统」替代临时 Prompt 的 AI 角色仓库
人工智能