在服务器上训练的网络放到本地台式机进行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设计】。
From Python正则表达式 re.match(pattern, string, flags=0) 尝试从字符串起始位置匹配一个模式；如果不是起始位置匹配成功，则 re.match(...
本文将原始的numpy array数据在pytorch下封装为Dataset类的数据集，为后续深度网络训练提供数据。 加载并保存图像信息 首先导入需要的库，定义各种路径。 impor...
线性回归实战 使用PyTorch定义线性回归模型一般分以下几步： 1.设计网络架构 2.构建损失函数(loss)和优化器(optimizer) 3.训练(包括前馈(forward...
复制代码 代码如下:# -*- coding: utf-8 -*-import sysfrom PyQt4 import QtCore, QtGuiclass MyWindow(QtGu...
本文实例讲述了Python数据分析之双色球统计两个红和蓝球哪组合比例高的方法。分享给大家供大家参考，具体如下： 统计两个红球和蓝球，哪个组合最多，显示前19组数据 #!/usr/bi...