在服务器上训练的网络放到本地台式机进行infer，结果出现报错：
AssertionError: Invalid device id
仔细检查后发现原来服务器有多个GPU，当时开启了两个进行加速运算。
net1 = nn.DataParallel(net1, device_ids=[0, 1])
而本地台式机只有一个GPU，调用数量超出所以报错。
改为
net1 = nn.DataParallel(net1, device_ids=[0])
问题解决。
以上这篇解决pytorch报错:AssertionError: Invalid device id的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
random是用于生成随机数的，我们可以利用它随机生成数字或者选择字符串。 •random.seed(x)改变随机数生成器的种子seed。 一般不必特别去设定seed，Pyt...
enumerate函数用于遍历序列中的元素以及它们的下标。 enumerate函数说明： 函数原型：enumerate(sequence, [start=0]) 功能：将可循环序列...
本文实现了PyQt5个各种弹出窗口：输入框、消息框、文件对话框、颜色对话框、字体对话框、自定义对话框 其中，为了实现自定义对话框的返回值，使用了信号/槽 本文基于 windows 7 +...
这篇文章主要介绍了Python csv模块使用方法代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 import csv d...
本文是一篇关于《Effective Python》书中一节的学习笔记，记录了示例代码和思路。 如果函数要产生一系列结果，那么最简单的做法就是把这些结果都放在一个列表里返回。 比如我们要查...