解决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中用keys()方法返回字典键的教程
May 21 Python
python实现简单淘宝秒杀功能
May 03 Python
Python 统计字数的思路详解
May 08 Python
[原创]Python入门教程5. 字典基本操作【定义、运算、常用函数】
Nov 01 Python
使用python的pexpect模块,实现远程免密登录的示例
Feb 14 Python
对python 调用类属性的方法详解
Jul 02 Python
浅谈Python_Openpyxl使用(最全总结)
Sep 05 Python
Python 实现递归法解决迷宫问题的示例代码
Jan 12 Python
python opencv实现信用卡的数字识别
Jan 12 Python
Python递归实现打印多重列表代码
Feb 27 Python
Python创建自己的加密货币的示例
Mar 01 Python
教你用python实现一个无界面的小型图书管理系统
May 21 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
我的论坛源代码(四)
2006/10/09 PHP
php下通过POST还是GET来传值
2008/06/05 PHP
php Memcache 中实现消息队列
2009/11/24 PHP
PHP的简易冒泡法代码分享
2012/08/28 PHP
浅析HTTP消息头网页缓存控制以及header常用指令介绍
2013/06/28 PHP
php实现图片添加水印功能
2014/02/13 PHP
PHP按行读取、处理较大CSV文件的代码实例
2014/04/09 PHP
PHP向socket服务器收发数据的方法
2015/01/24 PHP
php如何连接sql server
2015/10/16 PHP
javascript 学习笔记(八)javascript对象
2011/04/12 Javascript
JavaScript快速检测浏览器对CSS3特性的支持情况
2012/09/26 Javascript
jQuery文本框(input textare)事件绑定方法教程
2013/04/24 Javascript
jQuery浏览器CSS3特写兼容实例
2015/01/19 Javascript
在AngularJS应用中实现一些动画效果的代码
2015/06/18 Javascript
JS判断元素是否在数组内的实现代码
2016/03/30 Javascript
使用开源工具制作网页验证码的方法
2016/10/17 Javascript
微信小程序 保留小数(toFixed)详细介绍
2016/11/16 Javascript
详解MVC如何使用开源分页插件(shenniu.pager.js)
2016/12/16 Javascript
Node.js如何响应Ajax的POST请求并且保存为JSON文件详解
2017/03/10 Javascript
JavaScript实现简单评论功能
2017/08/17 Javascript
JS实现带动画的回到顶部效果
2017/12/28 Javascript
详解vue项目中如何引入全局sass/less变量、function、mixin
2018/06/02 Javascript
解决Vue.js由于延时显示了{{message}}引用界面的问题
2018/08/25 Javascript
d3.js实现图形拖拽
2019/12/19 Javascript
Python threading多线程编程实例
2014/09/18 Python
Python Scapy随心所欲研究TCP协议栈
2018/11/20 Python
Python网页正文转换语音文件的操作方法
2018/12/09 Python
详解用Python调用百度地图正/逆地理编码API
2020/07/02 Python
opencv 形态学变换(开运算,闭运算,梯度运算)
2020/07/07 Python
常用的HTML5列表标签
2017/06/20 HTML / CSS
信访工作者先进事迹
2014/01/17 职场文书
2014办公室副主任四风对照检查材料思想汇报
2014/09/20 职场文书
满月酒邀请函
2015/01/30 职场文书
思品教学工作总结
2015/08/10 职场文书
廉洁自律准则学习心得体会
2016/01/13 职场文书
Python3 使用pip安装git并获取Yahoo金融数据的操作
2021/04/08 Python