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

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

相关推荐
深蓝AI24 分钟前
12000 名工程师的高盛,只用四个月就收回了交易员的 AI 编程权限
人工智能·程序员
怕浪猫4 小时前
这个十几个agent 协作项目值得关注
人工智能·面试·github
~kiss~6 小时前
A Survey of On-Policy Distillation for LLMs 学习(一)
学习
starzy19906 小时前
深入理解Scala隐式类:扩展方法机制
开发语言·人工智能·scala
深蓝AI7 小时前
为什么 agent 的 demo 都很惊艳,上了生产就不靠谱
人工智能·程序员
数字新视界7 小时前
信创动环监控在机房环境管理中的智能化应用与实践分析
大数据·人工智能·物联网·信息与通信·动环监控系统
岁月宁静7 小时前
DeepSeek Harness 实战与架构解析:让 Agent 从“会聊”到“能交付”
人工智能·python·agent
深蓝AI8 小时前
一个项目五个 AI 同时干活,Claude Code 怎么让它们不互相打架
人工智能·程序员
forest_098 小时前
多 Agent 协作第二课:交叉验证≠独立验证
人工智能·python·算法·机器学习·prompt·agi
小杨小杨、努力变强!8 小时前
python学习笔记
笔记·学习