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

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

相关推荐
阳光宅男@李光熠几秒前
【电子通识】PCB外观检查机AVI(Automated Visual Inspection)
笔记·学习
牛油果子哥q4 分钟前
Prompt工程核心落地:结构化约束、Few-Shot示例、输出稳定性控
大数据·人工智能·prompt
所念皆星海9118 分钟前
Python学习---DAY10函数
开发语言·python·学习
刘天远8 分钟前
企业 Agent 需求怎么写:数据结构、流程图与 Python 校验
数据结构·人工智能·python·流程图
一次旅行8 分钟前
2026‑09‑05 AI产业深度解读|Claude 11天验证费马大定理、智能体Wiki通信逃逸、GPT-6 Astra安全攻防升级
人工智能·gpt·安全
人工智能AI技术12 分钟前
客服数据分析Agent落地实战:告别Demo,打通从零散对话到钉钉自动交付全链路
人工智能
wy2002031413 分钟前
小机构怎么新增学习能力项目
学习
新知图书14 分钟前
7.3 LangGraph中TypedDict与BaseModel的共通与嵌套
人工智能·智能体
旖旎夜光17 分钟前
【LangChain实战】LangChain 学习笔记(一):从定义大模型到工具调用
人工智能·笔记·python·学习·langchain
月华路20 分钟前
《模型不玄学》第16章 概率校准
人工智能·深度学习·机器学习