解决pytorch报错:AssertionError: Invalid device id的问题
作者：学EE的粒粒 发布时间：2021-05-15 16:13:42
标签：pytorch,AssertionError
在服务器上训练的网络放到本地台式机进行infer，结果出现报错：
AssertionError: Invalid device id
仔细检查后发现原来服务器有多个GPU，当时开启了两个进行加速运算。
net1 = nn.DataParallel(net1, device_ids=[0, 1])
而本地台式机只有一个GPU，调用数量超出所以报错。
改为
net1 = nn.DataParallel(net1, device_ids=[0])
问题解决。
来源：https://blog.csdn.net/baoyongshuai1509/article/details/85254298
0
投稿猜你喜欢
- W3C（万维网联盟）日前发布了HTML 5公共草案，这是10年来对HTML的一次主要升级。据国外媒体报道，W3C HTML5草案于本周二发布
- PHP的isset()函数 一般用来检测变量是否设置 格式：bool isset ( mixed var [, mixed var [, .
- 本次分析基于 CPython 解释器，python3.x版本在python2时代，整型有 int 类型和 long 长整型，长整型不存在溢出
介绍我们可以通过控制HeaderStyle, RowStyle, AlternatingRowStyle和其他一些属性来改变GridView- 本文实例讲述了python实现可将字符转换成大写的tcp服务器。分享给大家供大家参考。具体分析如下：下面的python代码执行后通过tcp监
- 本文实例讲述了正则表达式验证IPV4地址功能。分享给大家供大家参考，具体如下：IPV4地址由4个组数字组成，每组数字之间以.分隔，每组数字的
- 前言多线程一般用于同时调用多个函数，cpu时间片轮流分配给多个任务。 优点是提高cpu的使用率，使计算机减少处理多个任务的总时间；缺点是如果
- 介绍我们可以通过for循环来迭代list、tuple、dict、set、字符串，dict比较特殊dict的存储不是连续的，所以迭代（遍历）出
- div+css实现圆角边框，在网络上查看了一下，很多都是实现圆角的矩形的方法，我在这里介绍的是实现圆角矩形边框的方法。用代码说明问题：<
- Oracle9i中提供强大的迁移功能，可以从多种数据库向Oracle迁移数据。Oracle新发行的迁移工具提供了从Access2
- 如何导入数据数据可能有各种格式，虽然常见的是HDFS，但是因为在Python爬虫中数据库用的比较多的是MongoDB，所以这里会重点说说如何
- 写在前面的话:此篇还是asp相关的,相信玩ASP的都有这个感觉,当数据有5万多条时-------just like音乐网,要调用最新的10条
- 反馈说在选择时间时会出现遮挡选择器的情况，阻碍操作 如下图1，需要修改xadmin 文件 ，在widgets.py --->第28行添
- 1.linux下启动mysql的命令：mysqladmin start/ect/init.d/mysql start (前面为mysql的安
- Hello，各位读者朋友们好啊，我是小张~这不国庆嘛，就把最近很火的一个韩剧《鱿鱼游戏》刷了下，这部剧整体剧情来说还是非常不错的，很值得一看
- 步骤如下： 1.使用torchvision加载并预处理CIFAR-10数据集、2.定义网络3.定义损失函数和优化器4.训练网络并更新网络参数
- 很多朋友使用Dreamweaver一段时间后，开始热衷于寻找各式各样的插件，追求各种各样的特效，而对于Dreamweaver中的基本功能反而
- defer介绍defer是golang的一个特色功能，被称为“延迟调用函数”。当外部函数返回后执行defer。类似于其他语言的 try… c
- *在起初pip install matplotlib时，主动安装到当时最新版本（matplotlib==3.3.2），在StackOverf
- 本文实例为大家分享了梅尔倒谱系数实现代码，供大家参考，具体内容如下""" @author: zoutai@fi
