在服务器上训练的网络放到本地台式机进行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设计】。
strip去除空格 s = ' abcd efg ' print(s.strip()) #去除所有空格 print(s.lstrip()) #去除左边空格 print(s.rs...
快速上手Django实现项目 近期公司在做1个海淘的项目,APP为pylot。由于时间比较赶,加上隔壁那哥们不在,只能自己挑大梁了。结果,当项目做出来之后,被领导狠狠的批了一顿,说怎么用...
1、列表（list） list是处理一组有序项目的数据结构，即你可以在一个列表中存储一个序列的项目。 列表中的项目应该包括在方括号中，这样Python就知道你是指明一个列表。一旦...
开发中偶尔需要判断网络的连通性，没有什么方法比 ping 更直接了当，通常检查网络情况都是运行命令ping www.baidu.com ，查看输出信息即可。 C:\Users>...
本文较为详细的介绍了python中raw_input的用法，使用raw_input 能够很方便的丛控制台读入数据。具体用法示例如下： 1.输入字符串 #13222319810101*...