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设计】。

返回列表

在Python2中datetime对象没有timestamp方法，不能很方便的生成epoch，现有方法没有处理很容易导致错误。关于Epoch可以参见时区与Epoch 0 Python中生...

本文实例讲述了Python分治法定义与应用。分享给大家供大家参考，具体如下：分治法所能解决的问题一般具有以下几个特征： 1) 该问题的规模缩小到一定的程度就可以容易地解决 2) 该问题...

本文实例讲述了Python3实现的判断回文链表算法。分享给大家供大家参考，具体如下：问题：请判断一个链表是否为回文链表。方案一：指针法 class Solution: de...

getopt模块用于抽出命令行选项和参数，也就是sys.argv命令行选项使得程序的参数更加灵活。支持短选项模式和长选项模式例如 python scriptname.py -...

对dataframe绘图并保存： ax = df.plot() fig = ax.get_figure() fig.savefig('fig.png') 可以制定列，对该列各取...

宜配屋