在服务器上训练的网络放到本地台式机进行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中，我们经常会遇到字符串的拼接问题，几乎任何一种编程语言，都把...
继承是面向对象编程的一个重要的方式，因为通过继承，子类就可以扩展父类的功能。 回忆一下Animal类层次的设计，假设我们要实现以下4种动物： ...
假设通过爬虫你获取到了北京2016年3,10月份每天白天的最高气温（分别位于列表a，b），那么此时如何寻找出气温随时间（天）变化的某种规律？ a = [11,17,16,11,12,11...
1.基于rbac的权限管理 RBAC（Role-Based Access Control，基于角色的访问控制），就是用户通过角色与权限进行关联。简单地说，一个用户拥有若干角色，一个角色拥...
描述： 如果iterable的任何元素为true，则返回true。如果iterable为空，则返回false。相当于： def any(iterable): for elemen...