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

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

相关推荐
kingcjh972 分钟前
2.1 vLLM-Omni + Wan2.1-T2V-1.3B测试数据
深度学习
NOCSAH4 分钟前
生产大屏与统计报表:数智一体化平台如何实现工厂管理
人工智能·信息可视化·数智化一体平台·统好ai
BlockWay7 分钟前
从展台到策略执行:WEEX 在 NBX2026 展示 AI 真实应用场景
大数据·人工智能
不早睡不改名@11 分钟前
Netty源码解析---FastThreadLocal-addToVariablesToRemove方法详解
java·网络·笔记·学习·netty
CDA数据分析师干货分享16 分钟前
【访谈】食品专业转行数据分析师,CDA数据分析师二级备考经验
学习·信息可视化·数据分析·cda证书·cda数据分析师
zhongerzixunshi19 分钟前
一篇文章全面了解智能制造能力成熟度标准CMMM符合性评估
大数据·人工智能·制造
no245441020 分钟前
深度解析:WebP会在几年内取代JPG吗?
java·大数据·人工智能·科技·ai
龙文浩_23 分钟前
AI深度学习中的张量计算理论与实践
人工智能·神经网络
盐城吊霸天23 分钟前
Spring AI + Flux/FluxSink + SSE 实战技术笔记
人工智能·笔记·spring
软件算法开发25 分钟前
基于霸王龙优化算法的LSTM网络模型(TROA-LSTM)的一维时间序列预测matlab仿真
人工智能·matlab·lstm·一维时间序列预测·霸王龙优化·troa-lstm