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

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

相关推荐
高洁012 分钟前
大模型Prompt实战:精准生成专业技术文档
人工智能·python·数据挖掘·transformer·知识图谱
实在智能RPA2 分钟前
Agent如何帮助企业提升数据决策能力?2026年企业级智能体架构与落地实操深度解析
人工智能·ai·架构
ZPC82102 分钟前
手柄替代键盘
人工智能·算法·性能优化·机器人
Peter·Pan爱编程11 分钟前
OpenClaw进阶实战(二):本地模型量化与加速——GGUF、vLLM、OpenVINO实战
人工智能·openvino·vllm
CSDN官方博客13 分钟前
CSDN Bot配置教程
人工智能
龙萱坤诺14 分钟前
告别抽卡式生成:通义万相Wan2.7-Image实现“可控式创作
人工智能·通义千问·wan2.7-image
明月醉窗台29 分钟前
[Jetson] NVIDIA Jetson 全系列边缘计算芯片进阶指南
人工智能·边缘计算·cuda·jetson
禹中一只鱼32 分钟前
【力扣热题100学习笔记】 - 双指针
java·笔记·学习·leetcode·贪心算法
北京盟通科技官方账号34 分钟前
AGV/AMR 交付困局:如何在“极致成本”与“高可靠避障”之间寻找最优路径?
人工智能·数字孪生·agv·虚拟调试·工业现场·fe.screen-sim·fee