编程 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 set集合类型操作总结

Nov 07 Python

Python XML RPC服务器端和客户端实例

Nov 22 Python

Python中正则表达式详解

May 17 Python

Python编程实现的简单Web服务器示例

Jun 22 Python

python json.loads兼容单引号数据的方法

Dec 19 Python

Python之循环结构

Jan 15 Python

Django models.py应用实现过程详解

Jul 29 Python

django项目简单调取百度翻译接口的方法

Aug 06 Python

如何理解python中数字列表

May 29 Python

Python导入数值型Excel数据并生成矩阵操作

Jun 09 Python

基于python实现MQTT发布订阅过程原理解析

Jul 27 Python

让你相见恨晚的十个Python骚操作

Nov 18 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

PHP5中虚函数的实现方法分享

2011/04/20 PHP

PHP 自动加载的简单实现(推荐)

2016/08/12 PHP

PHP实现基于回溯法求解迷宫问题的方法详解

2017/08/17 PHP

js jquery分别实现动态的文件上传操作按钮的添加和删除

2014/01/13 Javascript

js实现仿微博滚动显示信息的效果

2015/12/21 Javascript

JavaScript html5 canvas绘制时钟效果

2016/03/01 Javascript

前端面试题及答案整理(二)

2016/08/26 Javascript

javascript中数组（Array)对象和字符串（String)对象的常用方法总结

2016/12/15 Javascript

Vue.js 2.0中select级联下拉框实例

2017/03/06 Javascript

js实现数组和对象的深浅拷贝

2017/09/30 Javascript

js Dom实现换肤效果

2017/10/21 Javascript

Django模板继承 extend标签实例代码详解

2019/05/16 Javascript

JavaScript仿京东秒杀倒计时

2020/03/17 Javascript

js实现表格单列按字母排序

2020/08/12 Javascript

解决vant-UI库修改样式无效的问题

2020/11/03 Javascript

Python读取ini文件、操作mysql、发送邮件实例

2015/01/01 Python

在Python中处理列表之reverse()方法的使用教程

2015/05/21 Python

实例解析Python的Twisted框架中Deferred对象的用法

2016/05/25 Python

详解duck typing鸭子类型程序设计与Python的实现示例

2016/06/03 Python

解决Django模板无法使用perms变量问题的方法

2017/09/10 Python

python实现数独游戏 java简单实现数独游戏

2018/03/30 Python

Python向Excel中插入图片的简单实现方法

2018/04/24 Python

Python自定义函数实现求两个数最大公约数、最小公倍数示例

2018/05/21 Python

Python+OpenCV实现实时眼动追踪的示例代码

2019/11/11 Python

PyCharm MySQL可视化Database配置过程图解

2020/06/09 Python

python UIAutomator2使用超详细教程

2021/02/19 Python

澳大利亚Mocha官方网站：包、钱包、珠宝和配饰

2019/07/18 全球购物

NOTINO英国：在线购买美容和香水

2020/02/25 全球购物

畜牧兽医本科生个人的自我评价

2013/10/11 职场文书

自荐信怎么写呢？

2013/12/09 职场文书

精通CAD能手自荐书

2014/01/31 职场文书

民族团结先进个人材料

2014/02/05 职场文书

妇联领导班子剖析材料

2014/08/21 职场文书

离婚纠纷代理词

2015/05/23 职场文书

2015年全民创业工作总结

2015/07/23 职场文书

vue中使用mockjs配置和使用方式

2022/04/06 Vue.js