在服务器上训练的网络放到本地台式机进行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设计】。
NTP（Network Time Protocol）是由美国德拉瓦大学的David L. Mills教授于1985年提出，设计用来在Internet上使不同的机器能维持相同时间的一种通讯...
系统环境：VC6 + Python-2.5.4 1、下载Python-2.5.4源码。 2、解压，打开D:\Python-2.5.4\PC\VC6\pcbuild.dsw，编译，D:\P...
writelines()方法写入字符串序列到文件。该序列可以是任何可迭代的对象产生字符串，字符串为一般列表。没有返回值。 语法 以下是writelines()方法的语法：...
Django在做url设置时，如果有两个APP，那么再import时会出错 解决的方法有两种： 方法一：在每个APP里设置单独的url.py文件，最后在总的url再进行设置： 1.分别...
在zip标准中，对文件名的 encoding 用的不是 unicode，而可能是各种软件根据系统的默认字符集来采用（此为猜测），因此zipfile中根据文件 flag 检测的时候，只支持...