编程 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 相关文章推荐

Python3实现生成随机密码的方法

Aug 23 Python

wxPython窗口中文乱码解决方法

Oct 11 Python

Python连接mssql数据库编码问题解决方法

Jan 01 Python

python语言使用技巧分享

May 31 Python

深入探究Django中的Session与Cookie

Jul 30 Python

Python实现简单遗传算法(SGA)

Jan 29 Python

Python实现随机生成手机号及正则验证手机号的方法

Apr 25 Python

CentOS 7下安装Python3.6 及遇到的问题小结

Nov 08 Python

10分钟教你用Python实现微信自动回复功能

Nov 28 Python

python库matplotlib绘制坐标图

Oct 18 Python

python安装第三方库如xlrd的方法

Oct 31 Python

python接口自动化框架实战

Dec 23 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静态新闻列表自动生成代码

2007/06/14 PHP

PHP MemCached高级缓存配置图文教程

2010/08/05 PHP

PHP中对用户身份认证实现两种方法

2011/06/04 PHP

PHP计数器的实现代码

2013/06/08 PHP

zf框架的Filter过滤器使用示例

2014/03/13 PHP

php如何实现只替换一次或N次

2015/10/29 PHP

Open and Print a Word Document

2007/06/15 Javascript

jQuery 操作XML入门

2008/12/25 Javascript

jquery给图片添加鼠标经过时的边框效果

2013/11/12 Javascript

nodejs获取本机内网和外网ip地址的实现代码

2014/06/01 NodeJs

使用Meteor配合Node.js编写实时聊天应用的范例

2015/06/23 Javascript

解决jQuery uploadify在非IE核心浏览器下无法上传

2015/08/05 Javascript

一起学写js Calender日历控件

2016/04/14 Javascript

JS查找字符串中出现次数最多的字符

2016/09/05 Javascript

JS 动态判断PC和手机浏览器实现代码

2016/09/21 Javascript

妙用缓存调用链实现JS方法的重载

2018/04/30 Javascript

vue-cli3.0如何使用CDN区分开发、生产、预发布环境

2018/11/22 Javascript

jQuery移动端跑马灯抽奖特效升级版(抽奖概率固定)实现方法

2019/01/18 jQuery

深入探索VueJS Scoped CSS 实现原理

2019/09/23 Javascript

vuex state中的数组变化监听实例

2019/11/06 Javascript

详解Python中的静态方法与类成员方法

2017/02/28 Python

Python原始字符串与Unicode字符串操作符用法实例分析

2017/07/22 Python

Python程序员面试题你必须提前准备!(答案及解析)

2018/01/23 Python

python使用正则表达式来获取文件名的前缀方法

2018/10/21 Python

PyQt 实现使窗口中的元素跟随窗口大小的变化而变化

2019/06/18 Python

python实现引用其他路径包里面的模块

2020/03/09 Python

Python图像阈值化处理及算法比对实例解析

2020/06/19 Python

Opencv+Python识别PCB板图片的步骤

2021/01/07 Python

canvas实现手机的手势解锁的步骤详细

2020/03/16 HTML / CSS

C，C++的几个面试题小集

2013/07/13 面试题

消防器材管理制度

2014/01/28 职场文书

高中体育教学反思

2014/01/29 职场文书

劳动竞赛活动总结

2014/05/05 职场文书

就业证明函

2015/06/17 职场文书

仓库管理制度范本

2015/08/04 职场文书

Python循环之while无限迭代

2022/04/30 Python