编程 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使用xmlrpclib模块实现对百度google的ping功能

Jun 02 Python

python之从文件读取数据到list的实例讲解

Apr 19 Python

Python中的CSV文件使用"with"语句的方式详解

Oct 16 Python

matplotlib.pyplot绘图显示控制方法

Jan 15 Python

python opencv图片编码为h264文件的实例

Dec 12 Python

Django重设Admin密码过程解析

Feb 10 Python

解决Pycharm中恢复被exclude的项目问题(pycharm source root)

Feb 14 Python

Python计算指定日期是今年的第几天(三种方法)

Mar 26 Python

python使用梯度下降和牛顿法寻找Rosenbrock函数最小值实例

Apr 02 Python

python上传时包含boundary时的解决方法

Apr 08 Python

基于python判断字符串括号是否闭合{}[]()

Sep 21 Python

Python pandas读取CSV文件的注意事项（适合新手）

Jun 20 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中删除数组的第一个元素和最后一个元素的函数

2015/03/07 PHP

Symfony2创建页面实例详解

2016/03/18 PHP

PHP数组操作简单案例分析

2016/10/15 PHP

实现PHP中session存储及删除变量

2018/10/15 PHP

JS中Iframe之间传值的方法

2013/03/11 Javascript

利用javaScript实现点击输入框弹出窗体选择信息

2013/12/11 Javascript

jQuery实现按钮的点击全选/反选单选框/复选框文本框表单验证

2015/06/25 Javascript

BootStrap中按钮点击后被禁用按钮的最佳实现方法

2016/09/23 Javascript

Vue.js 2.0 移动端拍照压缩图片上传预览功能

2017/03/06 Javascript

Vue-router结合transition实现app前进后退动画切换效果的实例

2017/10/11 Javascript

vue安装和使用scss及sass与scss的区别详解

2018/10/15 Javascript

帮你彻底搞懂JS中的prototype、__proto__与constructor（图解）

2019/08/23 Javascript

layui 地区三级联动 form select 渲染的实例

2019/09/27 Javascript

Vue 中 filter 与 computed 的区别与用法解析

2019/11/21 Javascript

微信小程序实现列表左右滑动

2020/11/19 Javascript

Python编程中对文件和存储器的读写示例

2016/01/25 Python

深入理解Python3 内置函数大全

2017/11/23 Python

详解python上传文件和字符到PHP服务器

2017/11/24 Python

解决python matplotlib imshow无法显示的问题

2018/05/24 Python

Python OpenCV处理图像之滤镜和图像运算

2018/07/10 Python

Keras使用tensorboard显示训练过程的实例

2020/02/15 Python

Python递归求出列表(包括列表中的子列表)的最大值实例

2020/02/27 Python

Python3.8安装Pygame教程步骤详解

2020/08/14 Python

详解Python中的路径问题

2020/09/02 Python

Python Pivot table透视表使用方法解析

2020/09/11 Python

使用css3做0.5px的细线的示例代码

2018/01/18 HTML / CSS

HTML5几个设计和修改的页面范例分享

2015/09/29 HTML / CSS

澳大利亚药房在线：ThePharmacy

2017/10/04 全球购物

什么是Linux虚拟文件系统VFS

2015/08/25 面试题

什么是规则表达式

2012/05/03 面试题

计算机开发个人求职信范文

2013/09/26 职场文书

初中生个人学习的自我评价

2013/12/04 职场文书

工作会议方案

2014/05/21 职场文书

环境工程专业自荐信范文

2014/06/24 职场文书

启动Tomcat时出现大量乱码的解决方法

2021/06/21 Java/Android

Node实现搜索框进行模糊查询

2021/06/28 Javascript