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

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

相关推荐
侠客工坊13 小时前
大模型落地移动端:解析侠客工坊端侧 Agent 的零拷贝(Zero-Copy)屏幕感知与空间映射
android·人工智能
yuan1999713 小时前
OpenCV ViBe 运动检测算法实现
人工智能·opencv·算法
mooyuan天天13 小时前
AI大模型辅助Web渗透测试-TRAE智能体自动化解CTF题(命令执行 powershell)
人工智能·web安全·渗透测试·ctf·ai辅助渗透测试
AAIshangyanxiu13 小时前
基于R语言机器学习方法在生态经济学领域中的实践应用
人工智能·机器学习·r语言·生态经济学·经济学
kongba00713 小时前
学习COZE编程 / LangGraph 通用工作流项目 提示词模板
java·网络·学习
半页码书13 小时前
半结构化面试是什么?跟结构化面试有什么区别?
人工智能·面试·职场和发展·求职招聘·职场发展·远程工作
人工智能AI技术13 小时前
自动驾驶 Agent:环境感知→路径规划→车辆控制
人工智能
AI先驱体验官13 小时前
臻灵:数字人形象驱动新突破,NVIDIA开源PersonaPlex带来的技术变局
大数据·人工智能·深度学习·重构·开源·aigc
郝学胜-神的一滴13 小时前
激活函数:神经网络的「非线性灵魂」,让模型从“直线”走向“万能”
人工智能·pytorch·python·深度学习·神经网络·程序人生·机器学习
研究点啥好呢13 小时前
Github热榜项目推荐 | 学习与贡献是开源的意义
学习·开源·github