编程 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求相邻数的方法示例

Aug 18 Python

python3 实现一行输入,空格隔开的示例

Nov 14 Python

django-rest-swagger对API接口注释的方法

Aug 29 Python

Linux下升级安装python3.8并配置pip及yum的教程

Jan 02 Python

pytorch下使用LSTM神经网络写诗实例

Jan 14 Python

解决Tensorboard 不显示计算图graph的问题

Feb 15 Python

python中wheel的用法整理

Jun 15 Python

Python Tricks 使用 pywinrm 远程控制 Windows 主机的方法

Jul 21 Python

Python面向对象实现方法总结

Aug 12 Python

pandas 数据类型转换的实现

Dec 29 Python

利用python为PostgreSQL的表自动添加分区

Jan 18 Python

详解matplotlib中pyplot和面向对象两种绘图模式之间的关系

Jan 22 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中截取中文字符串的代码小结

2011/07/17 PHP

php foreach循环中使用引用的问题

2013/11/06 PHP

php自定文件保存session的方法

2014/12/10 PHP

php中执行系统命令的方法

2015/03/21 PHP

功能强大的PHP POST提交数据类

2016/07/15 PHP

JS 文件本身编码转换图文教程

2009/10/12 Javascript

jQuery动画效果-slideUp slideDown上下滑动示例代码

2013/08/28 Javascript

Knockout数组(observable)使用详解示例

2013/11/15 Javascript

关闭页面window.location事件未执行的原因及解决方法

2014/09/01 Javascript

JS+CSS实现六级网站导航主菜单效果

2015/09/28 Javascript

文件上传,iframe跨域数据提交的实现

2016/11/18 Javascript

angularjs中ng-attr的用法详解

2016/12/31 Javascript

微信小程序 ES6Promise.all批量上传文件实现代码

2017/04/14 Javascript

jquery仿微信聊天界面

2017/05/06 jQuery

AngularJs定时器$interval 和 $timeout详解

2017/05/25 Javascript

Bootstrap fileinput文件上传组件使用详解

2017/06/06 Javascript

jQuery Form插件使用详解_动力节点Java学院整理

2017/07/17 jQuery

详解VueJS 数据驱动和依赖追踪分析

2017/07/26 Javascript

JS实现按钮控制计时开始和停止功能

2017/07/27 Javascript

jQuery Migrate 插件用法实例详解

2019/05/22 jQuery

Bootstrap 时间日历插件bootstrap-datetimepicker配置与应用小结

2019/05/28 Javascript

搭建vscode+vue环境的详细教程

2020/08/31 Javascript

[02:49]2018DOTA2亚洲邀请赛主赛事决赛日战况回顾 Mineski鏖战5局夺得辉耀

2018/04/10 DOTA

python多线程并发实例及其优化

2019/06/27 Python

python rsync服务器之间文件夹同步脚本

2019/08/29 Python

通过Python编写一个简单登录功能过程解析

2019/09/04 Python

Python生成器实现简单"生产者消费者"模型代码实例

2020/03/27 Python

Python利用PyPDF2库获取PDF文件总页码实例

2020/04/03 Python

Blue Nile蓝色尼罗河香港官网：世界最大在线钻石珠宝销售商

2020/05/07 全球购物

单位工程竣工验收方案

2014/03/16 职场文书

连带责任保证书

2014/04/29 职场文书

2014年公务员个人工作总结

2014/11/22 职场文书

2014年企业员工工作总结

2014/12/09 职场文书

漫画「狩龙人拉格纳」公开TV动画预告图

2022/03/22 日漫

大型强子对撞机再次重启探索“第五种自然力”

2022/04/29 数码科技

Python 文字识别

2022/05/11 Python