摘要:
具体问题报错信息为: 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)
浙公网安备 33010602011771号