解决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
投稿猜你喜欢
- 在许多人看来，HTML应该是WEB制作所有语言中最简单的语言，因为它不需要编译、封闭等，甚至只需要一个记事本就可以让其在浏览器中呈现出来。所
- 1、mysql下载下载地址：https://dev.mysql.com/downloads/mysql/5.6.html<br>
- 本文实例讲述了Python实现压缩文件夹与解压缩zip文件的方法。分享给大家供大家参考，具体如下：直接上代码#coding=utf-8#甄码
使用python实现文件导入，具体方法如下：文件样例可以自己random这里的temp1根据每一行的分隔符来读入，‘\n'表述回车t- 一、excel的内容二、效果三、需要用的库：openpyxlsmptlibemail.mime.textemail.header四、实现步骤
- 同样是取10条数据 select * from yanxue8_visit limit 10000,10 和 select * from y
- 问题：简单版连连看小游戏一个分割成w*h个正方格子的矩形板上，每个正方格子可以有游戏卡，也可以没有游戏卡两个游戏卡之间有一条路径相连需满足以
- 带你了解CGO编程大学时最开始学的语言莫过于C/C++,C/C++经过几十年的发展，已经积累了庞大的软件资产，它们很多久经考验而且性能已经足
- jupyter notebook安装完成之后需要密码，还有某些情况下也会出现需要输入密码的情况解决方法如下：1、在运行界面输入 jupyte
- 数据库 属性 &nbs
- 注意:安装时要保证Oracle安装目录不能带有中文字符（如果第一次安装出现“加载数据库错误areasQueries”的错误，一般是因为Ora
- 本文阐述了网页设计中的十大诀窍，作为网页开发者我们应该注意，这样才能最大限度得发挥网站的功能和效益： 预早筹划设计主页未必很艰难。
- 本文实例讲述了JS+HTML5实现上传图片预览效果。分享给大家供大家参考，具体如下：在项目中遇到用input标签file类型的文件上传，想实
- 本文通过图文并茂的方式给大家展示SQL SERVER数据库表记录只保留N天图文教程，具体方法步骤请看下文：第一步：首先设置SQL Serve
- 首先:我们介绍一下socket什么是socket:1. socket 在操作系统中它是处于应用层与传输层的抽象层,它是一组操作起来非常简单的
- 问题今天用nohup后台执行python程序，并将标准输出和错误输出重定向到一个log文件，但发现log文件隔好久才会更新，很煎熬。。。然而
- 前言selenium处理文件上传大致会有两种情况，一种是文件上传使用的是input标签元素，即<input type="fi
- 数据库优化是一项很复杂的工作，因为这最终需要对系统优化的很好理解才行。尽管对系统或应用系统的了解不多的情况下优化效果还不错，但是如果想优化的
- Python# coding=utf-8import timeimport sysdef func(): # 格式化输出时间 &
- 前言备受期待的django 2.0已经发布了，最大的一个变化就是不再支持python2.x版本了，这也为我们还在保守使用的2.x的同学们敲响
