首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 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 相关文章推荐
win10系统中安装scrapy-1.1
Jul 03 Python
Python结合ImageMagick实现多张图片合并为一个pdf文件的方法
Apr 24 Python
基于python log取对数详解
Jun 08 Python
Python3连接SQLServer、Oracle、MySql的方法
Jun 28 Python
TensorFlow利用saver保存和提取参数的实例
Jul 26 Python
Python实现的读取文件内容并写入其他文件操作示例
Apr 09 Python
浅谈Pandas Series 和 Numpy array中的相同点
Jun 28 Python
Python 余弦相似度与皮尔逊相关系数 计算实例
Dec 23 Python
python同义词替换的实现（jieba分词）
Jan 21 Python
深入浅析Python 命令行模块 Click
Mar 11 Python
django迁移文件migrations的实现
Mar 31 Python
Python socket服务常用操作代码实例
Jun 22 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
面向对象(3) hmac(1) hashlib(1) mt-swipe(1) 相对路径(1) requests(1) 相对路径 (1) openpyxl(1) 解密(1) Mint UI(1)
You might like
PHP Pear 安装及使用
2009/03/19 PHP
php设计模式 Command(命令模式)
2011/06/26 PHP
深入解读php中关于抽象(abstract)类和抽象方法的问题分析
2014/01/03 PHP
php模拟用户自动在qq空间发表文章的方法
2015/01/07 PHP
什么是OneThink oneThink后台添加插件步骤
2016/04/13 PHP
PHP基于phpqrcode类生成二维码的方法示例详解
2020/08/07 PHP
Jquery实现三层遍历删除功能代码
2013/04/23 Javascript
利用js正则表达式验证手机号,email地址,邮政编码
2014/01/23 Javascript
jQuery+PHP实现微信转盘抽奖功能的方法
2016/05/25 Javascript
HTML的select控件美化
2017/03/27 Javascript
Django与Vue语法的冲突问题完美解决方法
2017/12/14 Javascript
微信小程序实现页面浮动导航
2020/01/08 Javascript
Vue跨域请求问题解决方案过程解析
2020/08/07 Javascript
[01:06]欢迎来到上海，TI9
2018/08/26 DOTA
[01:07:11]Secret vs Newbee 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/17 DOTA
Python上下文管理器和with块详解
2017/09/09 Python
浅谈numpy库的常用基本操作方法
2018/01/09 Python
使用python实现语音文件的特征提取方法
2019/01/09 Python
python爬虫模拟浏览器的两种方法实例分析
2019/12/09 Python
Python print不能立即打印的解决方式
2020/02/19 Python
python GUI库图形界面开发之PyQt5选项卡控件QTabWidget详细使用方法与实例
2020/03/01 Python
pytorch cuda上tensor的定义 以及减少cpu的操作详解
2020/06/23 Python
浅析Python 条件控制语句
2020/07/15 Python
Django URL参数Template反向解析
2020/11/24 Python
杭州龙健科技笔试题.net部分笔试题
2016/01/24 面试题
linux面试题参考答案（8）
2016/04/19 面试题
省级四好少年事迹材料
2014/01/25 职场文书
物流毕业生个人的自我评价
2014/02/13 职场文书
经典婚礼主持词
2014/03/13 职场文书
主题实践活动总结
2014/05/08 职场文书
大学生工作求职信
2014/06/23 职场文书
融资合作协议书范本
2014/10/17 职场文书
2014年社区卫生工作总结
2014/12/18 职场文书
入党转正申请自我鉴定
2019/06/25 职场文书
MySQL 常见存储引擎的优劣
2021/06/02 MySQL
app场景下uniapp的扫码记录
2022/07/23 Java/Android