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

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

返回列表

使用 django.utils.translation.gettext_lazy() 函数，使得其中的值只有在访问时才会被翻译，而不是在 gettext_lazy() 被调用时翻译。例...

Python 标准库中有很多实用的工具类，但是在具体使用时，标准库文档上对使用细节描述的并不清楚，比如 urllib2 这个 HTTP 客户端库。这里总结了一些 urllib2 库的使用...

如下所示： import java.text.DecimalFormat import com.alibaba.fastjson.JSON import com.donews.dat...

本文实例讲述了python3.6中@property装饰器的使用方法。分享给大家供大家参考，具体如下： 1、@property装饰器的使用场景简单记录如下：负责把一个方法变成属性...

今天使用python画了几个好玩的3D展示图，现在分享给大家。先贴上图片使用的python工具包为： from matplotlib import pyplot as pl...

宜配屋