编程 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中super的用法实例

May 28 Python

Python的Django框架中if标签的相关使用

Jul 15 Python

定制FileField中的上传文件名称实例

Aug 23 Python

解决pandas .to_excel不覆盖已有sheet的问题

Dec 10 Python

Python3.5实现的三级菜单功能示例

Mar 25 Python

Python利用scapy实现ARP欺骗的方法

Jul 23 Python

解决Django 在ForeignKey中出现 non-nullable field错误的问题

Aug 06 Python

Python代码生成视频的缩略图的实例讲解

Dec 22 Python

python GUI库图形界面开发之PyQt5控件QTableWidget详细使用方法与属性

Feb 25 Python

使用Numpy对特征中的异常值进行替换及条件替换方式

Jun 08 Python

python基于opencv实现人脸识别

Jan 04 Python

浅谈Python3中datetime不同时区转换介绍与踩坑

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

Windows下PHP的任意文件执行漏洞

2006/10/09 PHP

paypal即时到账php实现代码

2010/11/28 PHP

用PHP写的基于Memcache的Queue实现代码

2011/11/27 PHP

关于shopex同步ucenter的redirect问题,导致script不运行

2013/04/10 PHP

PHP处理postfix邮件内容的方法

2015/06/16 PHP

php打包网站并在线压缩为zip

2016/02/13 PHP

php微信分享到朋友圈、QQ、朋友、微博

2019/02/18 PHP

checkbox设置复选框的只读效果不让用户勾选

2013/08/12 Javascript

可编辑下拉框的2种实现方式

2014/06/13 Javascript

JS拖拽插件实现步骤

2015/08/03 Javascript

js中字符串编码函数escape()、encodeURI()、encodeURIComponent()区别详解

2016/04/01 Javascript

Bootstrap按钮下拉菜单组件详解

2016/05/10 Javascript

详解angular2封装material2对话框组件

2017/03/03 Javascript

js canvas实现放大镜查看图片功能

2017/06/08 Javascript

jQuery阻止事件冒泡实例分析

2018/07/03 jQuery

详解Angular中通过$location获取地址栏的参数

2018/08/02 Javascript

图文讲解vue的v-if使用方法

2019/02/11 Javascript

原生js基于canvas实现一个简单的前端截图工具代码实例

2019/09/10 Javascript

在layui框架中select下拉框监听更改事件的例子

2019/09/20 Javascript

JavaScript 自定义html元素鼠标右键菜单功能

2019/12/02 Javascript

js实现一款简单踩白块小游戏(曾经很火)

2019/12/02 Javascript

在Python中操作字符串之startswith()方法的使用

2015/05/20 Python

python实现中文分词FMM算法实例

2015/07/10 Python

python交互式图形编程实例（一）

2017/11/17 Python

Python numpy 点数组去重的实例

2018/04/18 Python

python中使用 xlwt 操作excel的常见方法与问题

2019/01/13 Python

python实现梯度法 python最速下降法

2020/03/24 Python

美津浓巴西官方网站：Mizuno巴西

2019/07/24 全球购物

俄罗斯建筑和装饰材料在线商店：Stroilandia

2020/07/25 全球购物

C#笔试题

2015/07/14 面试题

心理学培训心得体会

2016/01/22 职场文书

高中16字霸气押韵班级口号集锦！

2019/06/27 职场文书

python3读取文件指定行的三种方法

2021/05/24 Python

Apache POI的基本使用详解

2021/11/07 Servers

python模拟浏览器使用selenium进入好友QQ空间并留言

2022/04/12 Python

element tree树形组件回显数据问题解决

2022/08/14 Javascript