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

Pyramid将models.py文件的内容分布到多个文件的方法

Nov 27 Python

python3.3教程之模拟百度登陆代码分享

Jan 16 Python

python下实现二叉堆以及堆排序的示例

Sep 29 Python

pandas 使用apply同时处理两列数据的方法

Apr 20 Python

Python实现的计算器功能示例

Apr 26 Python

python实现烟花小程序

Jan 30 Python

python调用并链接MATLAB脚本详解

Jul 05 Python

python和c语言的主要区别总结

Jul 07 Python

Python 识别12306图片验证码物品的实现示例

Jan 20 Python

Pycharm和Idea支持的vim插件的方法

Feb 21 Python

Python发起请求提示UnicodeEncodeError错误代码解决方法

Apr 21 Python

用Python将GIF动图分解成多张静态图片

Jun 11 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

从一个不错的留言本弄的mysql数据库操作类

2007/09/02 PHP

php更改目录及子目录下所有的文件后缀的代码

2010/09/24 PHP

php文件上传表单摘自drupal的代码

2011/02/15 PHP

解析thinkphp中的M()与D()方法的区别

2013/06/22 PHP

thinkphp中session和cookie无效的解决方法

2014/12/19 PHP

php数据库操作model类(使用__call方法)

2016/11/16 PHP

PHP使用PDO、mysqli扩展实现与数据库交互操作详解

2019/07/20 PHP

疯掉了,尽然有js写的操作系统

2007/04/23 Javascript

jQuery 入门讲解1

2009/04/15 Javascript

基于jQuery的自动完成插件

2011/02/03 Javascript

angularJS结合canvas画图例子

2015/02/09 Javascript

javascript获得当前的信息的一些常用命令

2015/02/25 Javascript

Hammer.js+轮播原理实现简洁的滑屏功能

2016/02/02 Javascript

Bootstrap table使用方法详细介绍

2016/12/09 Javascript

jquery submit()不能提交表单的解决方法

2017/04/24 jQuery

jQuery Ajax使用FormData上传文件和其他数据后端web.py获取

2017/06/11 jQuery

详解javascript appendChild()的完整功能

2018/08/18 Javascript

详解如何制作并发布一个vue的组件的npm包

2018/11/10 Javascript

vue中的ref和$refs的使用

2018/11/22 Javascript

小程序红包雨的实现示例

2019/02/19 Javascript

JS实现的字符串数组去重功能小结

2019/06/17 Javascript

jQuery实现轮播图源码

2019/10/23 jQuery

Js图片点击切换轮播实现代码

2020/07/27 Javascript

[01:11:48]Fnatic vs IG 2018国际邀请赛小组赛BO2 第二场 8.17

2018/08/18 DOTA

Scrapy-redis爬虫分布式爬取的分析和实现

2017/02/07 Python

浅谈scrapy 的基本命令介绍

2017/06/13 Python

pandas 数据归一化以及行删除例程的方法

2018/11/10 Python

使用python读取.text文件特定行的数据方法

2019/01/28 Python

python opencv图像处理(素描、怀旧、光照、流年、滤镜原理及实现)

2020/12/10 Python

python单例模式的应用场景实例讲解

2021/02/24 Python

买房子个人收入证明

2014/01/16 职场文书

网页美工求职信范文

2014/04/17 职场文书

幼儿园家长安全责任书

2014/07/22 职场文书

明确岗位职责

2015/02/14 职场文书

《秋天的图画》教学反思

2016/02/19 职场文书

2021年国漫热度排行前十，完美世界上榜，第四是美国动画作品

2022/03/18 国漫