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

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

相关推荐
打工仔折腾 AI2 分钟前
260+异构同步链路实战:如何用语义化自动翻译破解CDC DDL同步痛点
人工智能·后端·python·自然语言处理·性能优化·langchain·机器翻译
Eloudy9 分钟前
ASTRA-sim 3.0:通过高保真GPU和基础设施建模进行下一级分布式机器学习模拟
人工智能·分布式·机器学习
william_yangshun12 分钟前
【AI实战】OpenMausBot中文版使用教程:把一队Claude/Codex智能体塞进聊天窗口
人工智能
工业涂料百问13 分钟前
固化温度降 20℃ 的三重账:能耗、产能与涂层内应力
人工智能
梦醒江南14 分钟前
Agent 卡在网页上时,为什么不能换个真实入口继续?
人工智能·agent
流烟默18 分钟前
MCP第 2 篇:Java 本地实战:官方 SDK + stdio + 单能力 Server
人工智能·mcp
xian_wwq19 分钟前
【学习笔记】Milvus 是什么?为什么 RAG 系统经常使用它?-14
笔记·学习·milvus
无障碍研究院19 分钟前
2026TADC|圆桌对话:AI的包容性、反歧视、向善
人工智能
派小心.21 分钟前
DAU和MAU怎么计算?先确定活跃事件、用户键和统计窗口
大数据·服务器·前端·人工智能·数据分析
诺鸭船长21 分钟前
AI硬件大白话扫盲指南:从入门到榨干
人工智能