解决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的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

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