在服务器上训练的网络放到本地台式机进行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 简介 Python 是一个高层次的结合了解释性、编译性、互动性和面向对象的脚本语言。 Python...
主要是：前序遍历、中序遍历、后序遍历、层级遍历、非递归前序遍历、非递归中序遍历、非递归后序遍历 复制代码 代码如下:#!/usr/bin/env python#-*- coding:ut...
最近写一个BootStrap页面...因为功能需要所以决定一个页面解决所有问题，然后用jQuery来动态显示功能....然而这样做的话页面会相当庞大，一堆隐藏模态窗口和功能div都堆在一...
python中eval函数的用法十分的灵活，这里主要介绍一下它的原理和一些使用的场合。 下面是从python的官方文档中的解释： The arguments ar...
sequence 序列 sequence(序列)是一组有顺序的元素的集合 (严格的说，是对象的集合，但鉴于我们还没有引入“对象”概念，暂时说元素) 序列可以包含一个或多个元素，也可以没有...