编程 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的Django框架中settings文件的部署建议

May 30 Python

Python实现TCP/IP协议下的端口转发及重定向示例

Jun 14 Python

python3.5仿微软计算器程序

Mar 30 Python

Python实现读写sqlite3数据库并将统计数据写入Excel的方法示例

Aug 07 Python

Python用csv写入文件_消除空余行的方法

Jul 06 Python

python 以16进制打印输出的方法

Jul 09 Python

详解Django的model查询操作与查询性能优化

Oct 16 Python

python 杀死自身进程的实现方法

Jul 01 Python

深入了解Python枚举类型的相关知识

Jul 09 Python

python创建ArcGIS shape文件的实现

Dec 06 Python

Python如何把字典写入到CSV文件的方法示例

Aug 23 Python

python之json文件转xml文件案例讲解

Aug 07 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

40年前的这部特摄片恐龙特级克塞号80后的共同回忆

2020/03/08 日漫

给初学者的30条PHP最佳实践(荒野无灯)

2011/08/02 PHP

PHP中一些可以替代正则表达式函数的字符串操作函数

2014/11/17 PHP

PHP实现懒加载的方法

2015/03/07 PHP

PHP基于面向对象封装的分页类示例

2019/03/15 PHP

js实现的网站首页随机公告随机公告

2007/03/14 Javascript

JavaScript中的History历史对象

2008/01/16 Javascript

不用锚点也可以平滑滚动到页面的指定位置实现代码

2013/05/08 Javascript

解析js中获得父窗口链接getParent方法以及各种打开窗口的方法

2013/06/19 Javascript

javascript 通用loading动画效果实例代码

2014/01/14 Javascript

自编jQuery插件实现模拟alert和confirm

2014/09/01 Javascript

jquery制作属于自己的select自定义样式

2015/11/23 Javascript

javascript实现unicode与ASCII相互转换的方法

2015/12/10 Javascript

EasyUI的doCellTip实现鼠标放到单元格上提示单元格内容

2016/08/24 Javascript

Vue-cli 使用json server在本地模拟请求数据的示例代码

2017/11/02 Javascript

微信小程序使用canvas的画图操作示例

2019/01/18 Javascript

浅谈Vue.js组件(二)

2019/04/09 Javascript

ES6的异步操作之promise用法和async函数的具体使用

2019/12/06 Javascript

node+multer实现图片上传的示例代码

2020/02/18 Javascript

Vue通过阿里云oss的url连接直接下载文件并修改文件名的方法

2020/12/25 Vue.js

vue使用echarts画组织结构图

2021/02/06 Vue.js

python3.3实现乘法表示例

2014/02/07 Python

Python使用SocketServer模块编写基本服务器程序的教程

2016/07/12 Python

Python3.6使用tesseract-ocr的正确方法

2018/10/17 Python

基于YUV 数据格式详解及python实现方式

2019/12/09 Python

python super()函数的基本使用

2020/09/10 Python

美国渔具店：FishUSA

2019/08/07 全球购物

美国高端牛仔品牌：Silver Jeans

2019/12/12 全球购物

中专药剂专业应届毕的自我评价

2013/12/27 职场文书

环保专业大学生职业规划设计

2014/01/10 职场文书

社区敬老月活动实施方案

2014/02/17 职场文书

村委会贫困证明范文

2014/09/21 职场文书

2014小学语文教师个人工作总结

2014/12/03 职场文书

公路施工安全责任书

2015/05/08 职场文书

2015年艾滋病防治工作总结

2015/05/22 职场文书

刑事附带民事代理词

2015/05/25 职场文书