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

yipeiwu_com6年前 (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设计】。

返回列表

sae中安装有python环境,想让sae导入自己下载的django或者其他模块,可以在svn中新建一个文件目录,比如site-packages,跟python安装目录一样,这个目录存放...

不知道大家有没有这样的体验，windows电脑上查看一张gif图，默认就把IE给打开了，还弹出个什么询问项，好麻烦的感觉。所以为了解决自己的这个问题，写了个简单的文件夹内图片浏览工具。...

前言做网络爬虫的同学肯定见过各种各样的验证码，比较高级的有滑动、点选等样式，看起来好像挺复杂的，但实际上它们的核心原理还是还是很清晰的，本文章大致说明下这些验证码的原理以及带大家实现一...

有了ORM，我们就可以把Web App需要的3个表用Model表示出来： import time, uuid from transwarp.db import next_id fr...

平时当我们需要爬取一些我们需要的数据时，总是有些网站禁止同一IP重复访问，这时候我们就应该使用代理IP，每次访问前伪装自己，让“敌人”无法察觉。 oooooooooooooooOK，让我...

宜配屋