编程 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实现逆波兰计算表达式实例详解

May 06 Python

Python sys.argv用法实例

May 28 Python

Python实现批量转换文件编码的方法

Jul 28 Python

使用Python简单的实现树莓派的WEB控制

Feb 18 Python

python urllib urlopen()对象方法/代理的补充说明

Jun 29 Python

win7+Python3.5下scrapy的安装方法

Jul 31 Python

Python读取YUV文件,并显示的方法

Dec 04 Python

python 使用poster模块进行http方式的文件传输到服务器的方法

Jan 15 Python

python全栈知识点总结

Jul 01 Python

python实现登录密码重置简易操作代码

Aug 14 Python

Django框架HttpRequest对象用法实例分析

Nov 01 Python

Python数据类型最全知识总结

May 31 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类的封装与继承详解

2015/09/29 PHP

Javascript操纵Cookie实现购物车程序

2006/11/23 Javascript

得到文本框选中的文字，动态插入文字的js代码

2007/03/07 Javascript

?牟┛途W扣了一??效果出?? target=

2007/05/27 Javascript

jquery 图片轮换效果

2010/07/29 Javascript

javascript instanceof 内部机制探析

2010/10/15 Javascript

AJAX分页的代码(后台asp.net)

2011/02/14 Javascript

js下拉框二级关联菜单效果代码具体实现

2013/08/03 Javascript

Javascript基础教程之break和continue语句

2015/01/18 Javascript

JavaScript实现打开链接页面的方式汇总

2016/06/02 Javascript

javascript 注释代码的几种方法总结

2017/01/04 Javascript

react-redux中connect的装饰器用法@connect详解

2018/01/13 Javascript

angularJs在多个控制器中共享服务数据的方法

2018/09/30 Javascript

Vue实现渲染数据后控制滚动条位置(推荐)

2019/12/09 Javascript

vue开发中遇到的问题总结

2020/04/07 Javascript

Vue+Bootstrap收藏（点赞）功能逻辑与具体实现

2020/10/22 Javascript

[01:11]steam端dota2实名认证操作流程视频

2021/03/11 DOTA

Python 列表(List)操作方法详解

2014/03/11 Python

Python实现的二维码生成小软件

2014/07/11 Python

用Python进行行为驱动开发的入门教程

2015/04/23 Python

为什么选择python编程语言入门黑客攻防给你几个理由!

2018/02/02 Python

python实现求两个字符串的最长公共子串方法

2018/07/20 Python

Python实现数据可视化看如何监控你的爬虫状态【推荐】

2018/08/10 Python

Python pyinotify模块实现对文档的实时监控功能方法

2018/10/13 Python

python numpy元素的区间查找方法

2018/11/14 Python

python判断链表是否有环的实例代码

2020/01/31 Python

会计专业毕业生自我鉴定

2013/10/29 职场文书

市场营销专业应届生自荐信

2014/06/19 职场文书

软件测试专业推荐信

2014/09/18 职场文书

工作服管理制度范本

2015/08/06 职场文书

小学班主任研修日志

2015/11/13 职场文书

《小乌鸦爱妈妈》教学反思

2016/02/19 职场文书

2019年干货：自我鉴定

2019/03/25 职场文书

golang 实现并发求和

2021/05/08 Golang

vue 实现弹窗关闭后刷新效果

2022/04/08 Vue.js

pnpm对npm及yarn降维打击详解

2022/08/05 Javascript