解决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的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中的zip函数使用示例
Jan 29 Python
Python的Scrapy爬虫框架简单学习笔记
Jan 20 Python
Python 制作糗事百科爬虫实例
Sep 22 Python
django用户注册、登录、注销和用户扩展的示例
Mar 19 Python
快速解决PyCharm无法引用matplotlib的问题
May 24 Python
浅谈django orm 优化
Aug 18 Python
Python3 pandas 操作列表实例详解
Sep 23 Python
使用Python进行防病毒免杀解析
Dec 13 Python
python logging 日志的级别调整方式
Feb 21 Python
解决python DataFrame 打印结果不换行问题
Apr 09 Python
Python Selenium实现无可视化界面过程解析
Aug 25 Python
python 实现非极大值抑制算法(Non-maximum suppression, NMS)
Oct 15 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 header()函数使用说明
2008/07/10 PHP
使用PHP实现蜘蛛访问日志统计
2013/07/05 PHP
php中常量DIRECTORY_SEPARATOR用法深入分析
2014/11/14 PHP
Laravel 5框架学习之路由、控制器和视图简介
2015/04/07 PHP
php使用QueryList轻松采集js动态渲染页面方法
2018/09/11 PHP
php+ajax 文件上传代码实例
2019/03/18 PHP
HR vs ForZe BO3 第一场 2.13
2021/03/10 DOTA
javascript 弹出窗口中是否显示地址栏的实现代码
2011/04/14 Javascript
JS/FLASH实现复制代码到剪贴板(兼容所有浏览器)
2013/05/27 Javascript
node.js中的path.isAbsolute方法使用说明
2014/12/08 Javascript
Hammer.js+轮播原理实现简洁的滑屏功能
2016/02/02 Javascript
jQuery无刷新上传之uploadify3.1简单使用
2016/06/18 Javascript
Javascript 引擎工作机制详解
2016/11/30 Javascript
vue项目动态设置页面title及是否缓存页面的问题
2018/11/08 Javascript
js replace替换字符串同时替换多个方法
2018/11/27 Javascript
vue实现跨域的方法分析
2019/05/21 Javascript
Vue状态模式实现窗口停靠功能(灵动、自由, 管理后台Admin界面)
2020/03/06 Javascript
Vue 事件的$event参数=事件的值案例
2021/01/29 Vue.js
[36:41]完美世界DOTA2联赛循环赛FTD vs Magma第一场 10月30日
2020/10/31 DOTA
[48:18]DOTA2-DPC中国联赛 正赛 RNG vs Dynasty BO3 第二场 1月29日
2021/03/11 DOTA
在Python中操作文件之read()方法的使用教程
2015/05/24 Python
关于反爬虫的一些简单总结
2017/12/13 Python
celery4+django2定时任务的实现代码
2018/12/23 Python
Python字符编码转码之GBK,UTF8互转
2020/02/09 Python
Python 中的pygame安装与配置教程详解
2020/02/10 Python
python能开发游戏吗
2020/06/11 Python
CSS3中颜色线性渐变实战
2015/07/18 HTML / CSS
详解HTML5常用的语义化标签
2019/09/27 HTML / CSS
巴西葡萄酒销售网站:Wine.com.br
2017/11/07 全球购物
利用异或运算实现两个无符号数的加法运算
2013/12/20 面试题
员工晚婚的请假条
2014/02/08 职场文书
捐款通知怎么写
2015/04/24 职场文书
2016中秋节晚会开场白
2015/11/26 职场文书
2016小学优秀教师先进事迹材料
2016/02/26 职场文书
2019通用版劳动合同范本!
2019/07/11 职场文书
使用Djongo模块在Django中使用MongoDB数据库
2021/06/20 Python