摘要: 具体问题报错信息为: missing keys in state_dict error in loading state_dict 诊断出错原因,采用了如下代码 # 获取当前模型的状态字典 current_state_dict = model.state_dict() # 获取保存的状态字典 sav 阅读全文
posted @ 2025-07-29 15:45 fourk-ty 阅读(143) 评论(0) 推荐(0)
摘要: 首先安装cuda和cudnn 看了pytorch,支持版本有cuda12.1,12.4和12.6 (这个是错误步骤!!!我写博客的时间节点(202507)只支持cu128,请跳过踩坑经历看后面) cuDNN官网下载地址:https://developer.nvidia.com/rdp/cudnn-a 阅读全文
posted @ 2025-07-29 14:01 fourk-ty 阅读(1938) 评论(0) 推荐(0)
摘要: 更新5090显卡后的一系列问题 【linux版】 问题描述:(时间戳:20250623)更换5090显卡后,由于新旧显卡的驱动版本不一致,使得更换后nvidia-smi就没有输出了。【ubuntu20.04系统】 Step1: 一开始考虑是显卡驱动的问题,所以直接就按照驱动更新的方式来搞了: 参考了 阅读全文
posted @ 2025-06-23 15:46 fourk-ty 阅读(1134) 评论(0) 推荐(0)