编程 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正则表达式去掉数字中的逗号(python正则匹配逗号)

Dec 25 Python

python实现爬虫下载美女图片

Jul 14 Python

Python标准库inspect的具体使用方法

Dec 06 Python

Python+tkinter使用80行代码实现一个计算器实例

Jan 16 Python

numpy matrix和array的乘和加实例

Jun 28 Python

python 数字类型和字符串类型的相互转换实例

Jul 17 Python

selenium+python环境配置教程详解

May 28 Python

Python一行代码解决矩阵旋转的问题

Nov 30 Python

利用matplotlib实现根据实时数据动态更新图形

Dec 13 Python

只需要这一行代码就能让python计算速度提高十倍

May 24 Python

详解Go语言运用广度优先搜索走迷宫

Jun 23 Python

python中 Flask Web 表单的使用方法

May 20 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判断字符以及字符串的包含方法属性

2008/08/30 PHP

163的邮件用phpmailer发送(实例详解)

2013/06/24 PHP

PHP使用pear自带的mail类库发邮件的方法

2015/07/08 PHP

mysql_escape_string()函数用法分析

2016/04/25 PHP

laravel使用Faker数据填充的实现方法

2019/04/12 PHP

在vs2010中调试javascript代码方法

2011/02/11 Javascript

THREE.JS入门教程(6)创建自己的全景图实现步骤

2013/01/25 Javascript

node.js中的buffer.write方法使用说明

2014/12/10 Javascript

60个很实用的jQuery代码开发技巧收集

2014/12/15 Javascript

NodeJS中Buffer模块详解

2015/01/07 NodeJs

javascript操作Cookie（设置、读取、删除）方法详解

2015/03/18 Javascript

使用RequireJS优化JavaScript引用代码的方法

2015/07/01 Javascript

jQuery实现购物车表单自动结算效果实例

2015/08/10 Javascript

javascript自动切换焦点控制效果完整实例

2016/02/02 Javascript

jQuery 添加样式属性的优先级别方法(推荐)

2017/06/08 jQuery

Node解决简单重复问题系列之Excel内容的获取

2018/01/02 Javascript

通过jquery toggleClass()属性制作文章段落更改背景颜色

2018/05/21 jQuery

微信小程序学习笔记之获取位置信息操作图文详解

2019/03/29 Javascript

CKeditor4 字体颜色功能配置方法教程

2019/06/26 Javascript

layui实现三级导航菜单

2019/07/26 Javascript

Vue 解决路由过渡动画抖动问题(实例详解)

2020/01/05 Javascript

[01:29:31]VP VS VG Supermajor小组赛胜者组第二轮 BO3第一场 6.2

2018/06/03 DOTA

python基础教程之获取本机ip数据包示例

2014/02/10 Python

Python+matplotlib+numpy绘制精美的条形统计图

2018/01/02 Python

Python面向对象之接口、抽象类与多态详解

2018/08/27 Python

对python打乱数据集中X,y标签对的方法详解

2018/12/14 Python

Pytorch 使用 nii数据做输入数据的操作

2020/05/26 Python

Python3爬虫中识别图形验证码的实例讲解

2020/07/30 Python

Python 微信公众号文章爬取的示例代码

2020/11/30 Python

详解CSS3中的box-sizing（content-box与border-box）

2019/04/19 HTML / CSS

美国家居装饰和豪华家具购物网站：One Kings Lane

2018/12/24 全球购物

会计电算化专业毕业生推荐信

2013/12/24 职场文书

中药学专业毕业生推荐信

2014/07/10 职场文书

“四风”查摆问题自我剖析材料

2014/09/27 职场文书

2014年度培训工作总结

2014/11/27 职场文书

自信主题班会

2015/08/14 职场文书