PyTorch使用cpu加载模型运算方式

yipeiwu_com7年前 (2020-03-06)Python基础

没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加载模型运算方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。

返回列表

周末在家，写了个小程序，用于将阿拉伯数字转换化大写中文。程序没经过任何优化，出没经过详细的测试，挂到网上，方便将来有需要的时候直接拿来用。 #!/usr/bin/python #-*...

如果刚装上ubuntu会发现自带了python2和python3，但是发现系统没有带pip，所以输入下面的命令分别给他们装上pip： sudo apt-get install pyt...

前言词云的使用相信大家已经不陌生了，使用很简单，直接调用wordcloud包就可以了。它的主要功能是根据文本词汇和词汇频率生成图片，从中可以直观的看出各个词汇所占比重。之前我们也介绍...

最近帮人做了个贪吃蛇的游戏（交作业用），很简单，界面如下：开始界面：游戏中界面：是不是很简单、朴素。（欢迎大家访问GitHub）游戏是基于PyGame框架制作的，程序核心逻...

本文实例为大家分享了python制作英文字典的具体代码，供大家参考，具体内容如下功能有添加单词，多次添加单词的意思，查询，退出，建立单词文件。 keys=[] dic={} def...

宜配屋