编程 Python

解决pytorch报错:AssertionError: Invalid device id的问题

Posted in Python onJanuary 10, 2020

在服务器上训练的网络放到本地台式机进行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的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

解决pytorch报错:AssertionError: Invalid device id的问题

- Author -

学EE的粒粒

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐

Python的净值数据接口调用示例分享

Mar 15 Python

Python在线运行代码助手

Jul 15 Python

Python随机生成数据后插入到PostgreSQL

Jul 28 Python

Python多线程编程之多线程加锁操作示例

Sep 06 Python

Python matplotlib通过plt.scatter画空心圆标记出特定的点方法

Dec 13 Python

对Python捕获控制台输出流的方法详解

Jan 07 Python

python3 pygame实现接小球游戏

May 14 Python

Python笔记之工厂模式

Nov 20 Python

python 实现从高分辨图像上抠取图像块

Jan 02 Python

Python reversed函数及使用方法解析

Mar 17 Python

利用Python自动化操作AutoCAD的实现

Apr 01 Python

python中os.path.join()函数实例用法

May 26 Python

python3中关于excel追加写入格式被覆盖问题(实例代码)

Jan 10 #Python

mac使用python识别图形验证码功能

Jan 10 #Python

python列表推导和生成器表达式知识点总结

Jan 10 #Python

pytorch的梯度计算以及backward方法详解

Jan 10 #Python

Python如何获取Win7，Win10系统缩放大小

Jan 10 #Python

python-OpenCV 实现将数组转换成灰度图和彩图

Jan 09 #Python

Python 实现将数组/矩阵转换成Image类

Jan 09 #Python

You might like

php处理复杂xml数据示例

2016/07/11 PHP

Laravel学习教程之本地化模块

2017/08/18 PHP

phpStorm+XDebug+chrome 配置详解

2019/04/01 PHP

几个比较经典常用的jQuery小技巧

2010/03/01 Javascript

JavaScript中将一个值转换为字符串的方法分析[译]

2012/09/21 Javascript

div拖拽插件——JQ.MoveBox.js(自制JQ插件)

2013/05/17 Javascript

类似php的js数组的in_array函数自定义方法

2013/12/27 Javascript

jquery数组封装使用方法分享(jquery数组遍历)

2014/03/25 Javascript

jQuery页面加载初始化常用的三种方法

2014/06/04 Javascript

用jquery实现动画跳到顶部和底部(这个比较简单)

2014/09/01 Javascript

初识Javascript小结

2015/07/16 Javascript

js中获取jsp表单中radio类型的值简单实例

2016/08/15 Javascript

Node.js开发教程之基于OnceIO框架实现文件上传和验证功能

2016/11/30 Javascript

js实现带缓动动画的导航栏效果

2017/01/16 Javascript

微信小程序 UI与容器组件总结

2017/02/21 Javascript

JS实现页面内跳转的简单代码

2017/09/03 Javascript

Swiper 4.x 使用方法(移动端网站的内容触摸滑动)

2018/05/17 Javascript

JavaScript设计模式之工厂模式简单实例教程

2018/07/03 Javascript

对layui中表单元素的使用详解

2018/08/15 Javascript

JS实现的获取银行卡号归属地及银行卡类型操作示例

2019/01/08 Javascript

Vue触发隐藏input file的方法实例详解

2019/08/14 Javascript

Jquery高级应用Deferred对象原理及使用实例

2020/05/28 jQuery

Angular处理未可知异常错误的方法详解

2021/01/17 Javascript

python中关于时间和日期函数的常用计算总结(time和datatime)

2013/03/08 Python

python基础教程之分支、循环简单用法

2016/06/16 Python

Python实现学校管理系统

2018/01/11 Python

Python小白必备的8个最常用的内置函数（推荐）

2019/04/03 Python

python3模拟实现xshell远程执行liunx命令的方法

2019/07/12 Python

基于python实现文件加密功能

2020/01/06 Python

python模拟实现斗地主发牌

2020/01/07 Python

详解Python中namedtuple的使用

2020/04/27 Python

python单元测试框架pytest的使用示例

2020/10/07 Python

python使用requests库爬取拉勾网招聘信息的实现

2020/11/20 Python

Vivo俄罗斯官方在线商店：中国智能手机品牌

2019/10/04 全球购物

特色蛋糕店创业计划书

2014/01/28 职场文书

Redis 哨兵集群的实现

2021/06/18 Redis