解决pytorch报错:AssertionError: Invalid device id的问题
更新时间：2020年01月10日 09:06:37 作者：学EE的粒粒
今天小编就为大家分享一篇解决pytorch报错:AssertionError: Invalid device id的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
在服务器上训练的网络放到本地台式机进行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函数默认参数使用的踩雷避坑详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-07-07
这篇文章主要介绍了Django配置跨域并开发测试接口,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
这篇文章主要介绍了Python yield 使用浅析,本文给出了多个使用实例来分析yield的使用方法,需要的朋友可以参考下2015-05-05
点赞这个功能是我们现在经常会遇到的一个功能，下面这篇文章主要给大家介绍了关于基于Django框架利用Ajax实现点赞功能的相关资料，文中通过示例代码介绍的非常详细，需要的朋友们下面随着小编来一起学习学习吧2018-08-08
这篇文章主要介绍了python requests库的使用，帮助大家更好的利用python进行爬虫，感兴趣的朋友可以了解下2021-01-01
这篇文章主要介绍了Python字符串的转义字符，转义字符是指，用一些普通字符的组合来代替一些特殊字符，由于其组合改变了原来字符表示的含义，下文相关资料需要的小伙伴可以参考一下2022-04-04
这篇文章主要介绍了Python迭代器和生成器定义与用法,结合实例形式详细分析了Python迭代器和生成器的概念、原理、定义、使用方法及相关操作注意事项,需要的朋友可以参考下2018-02-02
构建优秀的软件需要遵循特定的规则并执行行业标准,如何在真实的生产环境开发中体现呢？在这篇文章中,我将向您展示我在Python项目中设置的4种工具,以简化开发工作流程并执行一些最佳实践,这些工具帮助我提高了效率,节省了时间,希望你读完也能有所收获2024-01-01
这篇文章主要介绍了详解将Django部署到Centos7全攻略，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-09-09
本文主要介绍了Python中查找缺失值的三种方法,包括pandas库的isnull()方法、numpy库的isnan()方法和scikit-learn库的SimpleImputer类,感兴趣的可以了解一下2023-11-11
最新评论