编程 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中的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