在服务器上训练的网络放到本地台式机进行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设计】。
需求是在某一指定的时刻执行操作 网上的建议多为通过调用Scheduler的add_date_job实现 不过APScheduler 3.0.1与之前差异较大， 无法通过上述方法实现 参考...
本人在做项目的时候遇到一个问题： 某个函数需要在每个小时的 3 分钟时候被执行一次，我希望我 15：45 启动程序，过了18 分钟在 16：03 这个函数被执行一次，下一次过 60 分钟...
Python2.6 之前：字符串转换为整形和浮点型 >>>import string >>>string.atoi('34.1') 34 &...
Python 遍历 最近学习python这门语言，感觉到其对自己的工作效率有很大的提升，下面废话不多说，直接贴代码 #!/usr/bin/env python # -*- codin...
python中字符串数组如何逆序排列？下面给大家介绍几种方法： 1、数组倒序： 原始元素的倒序排列 （1）切片 >>> arr = [1,2,3,4,3,4]>...