在服务器上训练的网络放到本地台式机进行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设计】。
下面的demo是根据需求写的简单测试脚本 #!/usr/bin/env python # coding: utf-8 # 第一个列表为依赖组件和版本号，后面紧跟负责人名称 # 接着出...
Pandas中对 时间 这个属性的处理有非常非常多的操作。 而本文对其中一个大家可能比较陌生的方法进行讲解。其他的我会陆续上传。 应用情景是这样的：考虑到有一个数据集，数据集中有用户注...
Python 不建议用一行 import 所有模块，而是推荐用多行 import 来导入模块。更多的是出于操作便利性的原因，与代码本身无关： 更易于阅读 import fred 比...
本文实例讲述了python使用any判断一个对象是否为空的方法。分享给大家供大家参考。 具体实现代码如下： 复制代码 代码如下:>>> eth = {"eth0″:"1...
numpy库是Python进行数据分析和矩阵运算的一个非常重要的库，可以说numpy让Python有了matlab的味道 本文主要介绍几个numpy库下的小函数。 1、mat函数 mat...