编程 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中用keys()方法返回字典键的教程

May 21 Python

python实现简单淘宝秒杀功能

May 03 Python

Python 统计字数的思路详解

May 08 Python

[原创]Python入门教程5. 字典基本操作【定义、运算、常用函数】

Nov 01 Python

使用python的pexpect模块,实现远程免密登录的示例

Feb 14 Python

对python 调用类属性的方法详解

Jul 02 Python

浅谈Python_Openpyxl使用（最全总结）

Sep 05 Python

Python 实现递归法解决迷宫问题的示例代码

Jan 12 Python

python opencv实现信用卡的数字识别

Jan 12 Python

Python递归实现打印多重列表代码

Feb 27 Python

Python创建自己的加密货币的示例

Mar 01 Python

教你用python实现一个无界面的小型图书管理系统

May 21 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

我的论坛源代码(四)

2006/10/09 PHP

php下通过POST还是GET来传值

2008/06/05 PHP

php Memcache 中实现消息队列

2009/11/24 PHP

PHP的简易冒泡法代码分享

2012/08/28 PHP

浅析HTTP消息头网页缓存控制以及header常用指令介绍

2013/06/28 PHP

php实现图片添加水印功能

2014/02/13 PHP

PHP按行读取、处理较大CSV文件的代码实例

2014/04/09 PHP

PHP向socket服务器收发数据的方法

2015/01/24 PHP

php如何连接sql server

2015/10/16 PHP

javascript 学习笔记（八）javascript对象

2011/04/12 Javascript

JavaScript快速检测浏览器对CSS3特性的支持情况

2012/09/26 Javascript

jQuery文本框(input textare)事件绑定方法教程

2013/04/24 Javascript

jQuery浏览器CSS3特写兼容实例

2015/01/19 Javascript

在AngularJS应用中实现一些动画效果的代码

2015/06/18 Javascript

JS判断元素是否在数组内的实现代码

2016/03/30 Javascript

使用开源工具制作网页验证码的方法

2016/10/17 Javascript

微信小程序保留小数(toFixed)详细介绍

2016/11/16 Javascript

详解MVC如何使用开源分页插件（shenniu.pager.js）

2016/12/16 Javascript

Node.js如何响应Ajax的POST请求并且保存为JSON文件详解

2017/03/10 Javascript

JavaScript实现简单评论功能

2017/08/17 Javascript

JS实现带动画的回到顶部效果

2017/12/28 Javascript

详解vue项目中如何引入全局sass/less变量、function、mixin

2018/06/02 Javascript

解决Vue.js由于延时显示了{{message}}引用界面的问题

2018/08/25 Javascript

d3.js实现图形拖拽

2019/12/19 Javascript

Python threading多线程编程实例

2014/09/18 Python

Python Scapy随心所欲研究TCP协议栈

2018/11/20 Python

Python网页正文转换语音文件的操作方法

2018/12/09 Python

详解用Python调用百度地图正/逆地理编码API

2020/07/02 Python

opencv 形态学变换(开运算，闭运算，梯度运算)

2020/07/07 Python

常用的HTML5列表标签

2017/06/20 HTML / CSS

信访工作者先进事迹

2014/01/17 职场文书

2014办公室副主任四风对照检查材料思想汇报

2014/09/20 职场文书

满月酒邀请函

2015/01/30 职场文书

思品教学工作总结

2015/08/10 职场文书

廉洁自律准则学习心得体会

2016/01/13 职场文书

Python3 使用pip安装git并获取Yahoo金融数据的操作

2021/04/08 Python