pytorch——多GPU训练，CPU测试出现的问题_runtimeerror: module must have its parameters and -程序员宅基地

技术标签： pytorch

我在训练模型的时候使用了nn.DataParallel，并在在服务器gpu测试也同时将训练好的模型并行载入，这样没有问题，但是我将训练好的模型在本地cpu运行的时候出错。这里记录出错的地方，并记录大致流程。

1. CPU读取模型：

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

根据提示看出CPU读取模型的时候应该带map_location=torch.device('cpu')参数，所以加上参数：

load_model = torch.load(parser.model, map_location=torch.device('cpu'))

2. 模型载入

由于我在保存模型的时候直接把网路结构也一并保存进去，所以在读取预训练模型的时候就直接将模型读出。但是出现了这个错误：

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

于是我将模型的参数显示后看到如下：
在这里插入图片描述
在参数名称这个地方，相比原来的名字多了module.，所以只要把多出来的部分删去，并将新命名的参数字典重新载入即可。

本文链接：https://blog.csdn.net/weixin_38208912/article/details/104703729

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

1024 分辨率下最快模型，字节跳动文生图开放模型 SDXL-Lightning 发布_sdxllightning下载-程序员宅基地

文章浏览阅读1k次，点赞28次，收藏25次。很高兴跟大家分享我们最新的文生图模型 —— SDXL-Lightning，它实现了前所未有的速度和质量，并且已经向社区开放。_sdxllightning下载

关于cloacked-pixel的一些总结_03-cloacked-pixel-程序员宅基地

文章浏览阅读1.5k次，点赞2次，收藏2次。前两天遇到一道lsb隐写的题目，需要用到cloacked-pixel这个脚本。工具地址下载后解压即可，这里需要注意，该脚本是基于python2的！但我电脑anaconda里面只有python3并没有很好的python基础，但借助anaconda可以轻松解决很多问题！教程如下：anaconda中添加python2但是在运行脚本时还会提示缺库(注意运行时要activate pythonXX手动切换到你所配置的python2环境下)继续使用anaconda为python2装缺少的._03-cloacked-pixel

WPF DataGrid添加右键菜单-程序员宅基地

文章浏览阅读2k次，点赞2次，收藏2次。原文http://home.cnblogs.com/group/topic/54788.html用代码添加:View Code <DataGrid.ContextMenu> <ContextMenu Name="dgmenu1" StaysOpen="true"> &..._wpf datagrid右键菜单

EB Tresos Studio离线激活方法_ebtresos离线激活-程序员宅基地

文章浏览阅读1.9k次，点赞2次，收藏6次。EB Tresos Studio离线激活方法_ebtresos离线激活

B端系统-权限管理_如何将后台管理系统b端化-程序员宅基地

文章浏览阅读95次。当然用户组是可以拓展的，部门和职位常用在内部的管理系统，如果是面向c端的系统，比如淘宝网的商家，商家自身也有一套组织架构，如采购部，销售部，客服部，后勤部，有些人拥有客服权限，有些人拥有上架权限等，这就体现了用户组的扩展性。关于数据权限的处理，常见的有两种方式，一种是在角色内完成数据权限的定义，另一种是将角色和权限分开，两种方式各有偏重。即页面的功能按钮，包括查看，新增，修改，删除，审核等，用户点击删除按钮时，后台会校验用户角色下用户的所有权限是否包含该删除权限，如果是，就可以下一步，反之提示无权限。_如何将后台管理系统b端化

音乐节拍提取一-程序员宅基地

文章浏览阅读601次。前段时间倒腾了一下音乐节拍数检测，参考下面的网上的一个测试歌曲列表做了下对比，效果还不错，基本上都是准的。Itunes LinkNameTimeArtistBPMAlbumGenreAmazon LinkLoneliest Soul03:35Grace Potter and the Nocturnals168The Lion The Bea..._提取音乐节奏

随便推点

win11实时字幕无法下载问题_微软实时字幕下载不了-程序员宅基地

文章浏览阅读776次。一直卡着的话直接去Microsoft Store下载这个试试看。_微软实时字幕下载不了

如何让谷歌Chrome地址栏恢复显示“www”和“https://”标识符_chrome 地址显示原始-程序员宅基地

文章浏览阅读1.1w次。如何让谷歌Chrome地址栏恢复显示“www”和“https://”标识符地址栏隐藏“www”和“https://”标识符　　谷歌 Chrome 现在默认在所有网站地址栏中少了一些内容，“www”子域和“https://”被隐藏起来了，因为谷歌认为这些不是大多数人要关注的信息。　　Chrome 的产品经理 Emily Schechter 说，他们将开始从桌面版和 Android 版的第..._chrome 地址显示原始

目标跟踪数据集整理（四）----TColor-128(Temple Color 128)_encoding color information for visual tracking: al-程序员宅基地

文章浏览阅读3.5k次。文章目录Encoding Color Information for Visual Tracking:Algorithms and Benchmark 2015官网下载数据集（4.4G）本文认为颜色信息可以提供丰富的判别线索对于视觉推理，大多数现代视觉跟踪器限制在灰度域。（也就是主要解决输入序列是灰度版本）因此我们在算法和基准两方面做了系统的研究，证明了颜色信息可以帮助提升视觉跟踪效果。..._encoding color information for visual tracking: algorithms and benchmark