在服务器上训练的网络放到本地台式机进行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设计】。
很多时候我们会发现有的时候输出的结果特别多的时候，会在最后输出时用。。。代替，最后输出一个总长度，那要咋么弄咧？ import pandas as pd # 设置显示的最大列、宽等参...
本文实例讲述了Python数据结构之双向链表的定义与使用方法。分享给大家供大家参考，具体如下： 和单链表类似，只不过是增加了一个指向前面一个元素的指针而已。 示意图： python 实...
主要内容 1.函数基本语法及特性 2.参数与局部变 3.返回值 4.递归 5.名函数 6.函数式编程介绍 7.阶函数 8.内置函数 函数基本语法及特性 定义 数学函数定义:一般的，在一...
本文实例讲述了Python基于pygame实现的font游戏字体。分享给大家供大家参考，具体如下： 在pygame游戏开发中，一个友好的UI中，漂亮的字体是少不了的 今天就给大伙带来有关...
1、代码1： （1）进度条等显示在主窗口状态栏的右端，代码如下： from PyQt5.QtWidgets import QMainWindow, QProgressBar, QAp...