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

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 Agent 的记忆机制
人工智能
不想当程序汪的第N天4 分钟前
【AI】智能体经典范式
人工智能
武子康5 分钟前
删掉邮箱后,Agent Trace 仍可能泄露什么:一条可重放脱敏流水线
人工智能·llm·agent
Csvn6 分钟前
LLM 当裁判?先懂它的 4 个偏心——自动化评测实战(E03)
人工智能
呆萌很6 分钟前
简要介绍 torchvision.datasets.ImageFolder
人工智能
guanguan0_06 分钟前
用 AI 做技术方案评审:输入 3 个方案,输出对比矩阵 + 推荐理由
javascript·人工智能·矩阵·ai编程
代码里的AI星11 分钟前
深度解析:基于RAG架构的企业级“品牌AI可见度”监测体系构建
人工智能·架构
山岚的运维笔记11 分钟前
mysql 专业笔记 -- 第 1 章:MySQL 入门
运维·数据库·笔记·后端·学习·mysql·dba
YHL12 分钟前
🐴 Harness 工程:用工程化手段驯服 LLM 的幻觉
人工智能
陈彬深大14 分钟前
《AI 渐进编程》之二十八: 对 AI 的理解决定使用效果
人工智能