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

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

相关推荐
遥感知识服务1 小时前
不用灾前影像,AI能在40秒内画出一万平方公里洪水图吗?
人工智能
既然如此,那就开摆1 小时前
ULIP总结
笔记·学习
湘美书院--湘美谈教育1 小时前
湘美谈教育湘美书院成功学系列:标准即是文明,AI时代的走向
大数据·人工智能·深度学习·机器学习·生活
顿哥GPT1 小时前
7月更新 ChatGPT、Codex、Pro、Plus 背后的工程难题:AI Consistency 将成为 LLM-Native 系统的下一道门槛(5.6)
人工智能·chatgpt
FriendshipT1 小时前
Ultralytics:解读C3k2模块
人工智能·pytorch·python·深度学习·目标检测
技录局2 小时前
Comet 实现原理深度解析:如何让 AI 编程工作流可恢复、可约束、可评测
人工智能
武子康2 小时前
Hugging Face AI 驱动入侵真正暴露的是 Dataset Processing 信任边界(攻击链 + 三层隔离 + 行动清单)
人工智能·安全·llm
GuWenyue2 小时前
LLM工具调用慢到崩溃?1套Promise.all并行方案,Agent性能直接翻倍
人工智能
GuWenyue2 小时前
复刻Cursor编程Agent!4套文件命令工具+Promise.all并发,AI自动生成React项目
人工智能
技录局2 小时前
Superpowers 实现原理深度解析:如何把工程纪律变成 Agent 的默认行为
人工智能