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

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

相关推荐
Nayxxu1 小时前
Claude 长上下文调用成本分析:token、缓存、摘要与切片策略
人工智能
我材不敲代码1 小时前
目标检测:YOLOv12环境配置,全文详细教程
人工智能·yolo·目标检测
海盗12341 小时前
AI 每日要闻速递-2026年5月18日
人工智能
JavaAgent架构师1 小时前
Spring AI接入OpenAI报错401/429?3种原因+完整解决代码
人工智能·后端
RSTJ_16251 小时前
PYTHON+AI LLM DAY FOURTY-NINE
人工智能·python·深度学习
测试员周周1 小时前
【AI测试路线图2】功能测试转 AI 测试:4~5 个月,一条最稳的路
开发语言·人工智能·python·功能测试·测试工具·单元测试·pytest
05大叔1 小时前
生成式任务
人工智能·语音识别
fengfuyao9851 小时前
基于MATLAB的ALOHA防碰撞、二进制搜索算法和帧时隙算法
人工智能·算法·matlab
Hali_Botebie1 小时前
【蒸馏】Tinybert:Distilling BERT for natural language understanding.
人工智能·深度学习·bert
阳光九叶草LXGZXJ1 小时前
达梦数据库-学习-52-DmDrs参数介绍(Manager模块)
linux·运维·数据库·sql·学习