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

返回列表

本文实例讲述了python使用paramiko实现远程拷贝文件的方法。分享给大家供大家参考，具体如下：首先是安装paramiko库(其实现了SSH2安全协议)，ubuntu下可直接通过...

对于 PyTorch 的基本数据对象 Tensor (张量)，在处理问题时，需要经常改变数据的维度，以便于后期的计算和进一步处理，本文旨在列举一些维度变换的方法并举例，方便大家查看。维...

前言：前两天准备用 Python 在 Spark 上处理量几十G的数据，熟料在利用PyCharm进行PySpark远程调试时掉入深坑，特写此博文以帮助同样深处坑中的bigdata&mac...

一：字符串的方法与操作 *注意：首字母为l的为从左边操作，为r的方法为从右边操作 1.__contains__()判断是否包含判断指定字符或字符串是否包含在一个字符串内,返回值为tru...

1、层次索引 1.1 定义在某一个方向拥有多个（两个及两个以上）索引级别，就叫做层次索引。通过层次化索引，pandas能够以较低维度形式处理高纬度的数据通过层次化索引，可以按照层次...

宜配屋