在服务器上训练的网络放到本地台式机进行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设计】。
本文实例讲述了Python实现字符串反转的常用方法。分享给大家供大家参考，具体如下： 下面是实现python字符串反转的四种方法： 1. 切片 def rev(s): return...
起步 在Python中，对于一个对象的属性访问，我们一般采用的是点(.)属性运算符进行操作。例如，有一个类实例对象foo，它有一个name属性，那便可以使用foo.name对此属性进行...
树的定义树是一种重要的非线性数据结构，直观地看，它是数据元素（在树中称为结点）按分支关系组织起来的结构，很象自然界中的树那样。树结构在客观世界中广泛存在，如人类社会的族谱和各种社会组织机...
本文实例讲述了Python实现二维有序数组查找的方法。分享给大家供大家参考，具体如下： 题目：在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请...
简介 没有切换开销。因为子程序切换不是线程切换，而是由程序自身控制，没有线程切换的开销，因此执行效率高， 不需要锁机制。因为只有一个线程，也不存在同时写变量冲突，在协程中控制共享资源...