PyTorch使用cpu加载模型运算方式
更新时间：2020年01月13日 09:51:02 作者：ngy321
今天小编就为大家分享一篇PyTorch使用cpu加载模型运算方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
没gpu没cuda支持的时候加载模型到cpu上计算
将
model = torch.load(path, map_location=lambda storage, loc: storage.cuda(device))
改为
model = torch.load(path, map_location='cpu')
然后删掉所有变量后面的.cuda()方法
以上这篇PyTorch使用cpu加载模型运算方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇pytorch 实现在一个优化器中设置多个网络参数的例子，具有很好参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要给大家介绍了关于利用Python暴力破解zip文件口令的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2017-12-12
PySimpleGUI就是GUI工具包，用于创建GUI的。本文将利用PySimpleGUI编写一个GUI翻译器，感兴趣的小伙伴可以跟随小编一起动手尝试一下2022-08-08
这篇文章主要为大家详细介绍了chrome配置selenium操作的方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-03-03
PyQt为事件处理提供了两种机制高级的信号与槽机制，以及低级的事件处理机制，这篇文章主要介绍了PyQt5 事件处理机制,需要的朋友可以参考下2022-11-11
这篇文章主要介绍了关于Python的Thread线程模块详解,进程是程序的一次执行，每个进程都有自己的地址空间、内存、数据栈以及其他记录其运行的辅助数据,需要的朋友可以参考下2023-05-05
这篇文章主要介绍了Django+uwsgi+Nginx上线最佳实战，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-03-03
这篇文章主要介绍了如何对numpy 矩阵进行通道间求均值，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
这篇文章主要介绍了在Python里的dict和set的背后小秘密,dict字典是Python中的重要基础知识,set与其类似,需要的朋友可以参考下2021-11-11
这篇文章主要介绍了使用keras根据层名称来初始化网络，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
最新评论