在服务器上训练的网络放到本地台式机进行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设计】。
python2.7安装目录下没有Scripts文件夹。这种问题可能是你装的python安装包年代久远了，到官网下载最新的python2.7安装能解决这个问题。python2.7下载地址：...
pandas进行打印，控制台的显示默认是换行的在pycharm中的控制台也没有办法设置 可加入如下代码打印，结果不换行，看着数据更加直观 import pandas as pd i...
（1）lambda lambda是Python中一个很有用的语法，它允许你快速定义单行最小函数。类似于C语言中的宏，可以用在任何需要函数的地方。 基本语法如下： 函数名 = lambda...
测试环境 win10 python 3.5 yield功能简介 简单来说，yield 的作用就是把一个函数变成一个 generator，带有 yield 的函数不再是一个普通函数，P...
本文实例讲述了Python自定义装饰器原理与用法。分享给大家供大家参考，具体如下： 什么是装饰器？装饰器本质是一个函数，它可以在不改变原来的函数的基础上额外的增加一些功能。如常见的@cl...