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

返回列表

这篇文章主要介绍threading模块中的主类Thread的一些主要方法，实例代码如下：复制代码代码如下: ''' Created on 2012-9-7 ...

本文实例讲述了Python shutil模块用法。分享给大家供大家参考，具体如下： shutil模块主要作用与拷贝文件用的。 1.shutil.copyfileobj(文件1，文件2)...

这篇文章主要介绍了python基于celery实现异步任务周期任务定时任务,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 hello,...

首先到官网下载python2.7.3版本，编译安装复制代码代码如下: $wget http://www.python.org/ftp/python/2.7.3/Python-2.7....

举例说明 TensorFlow中的变量一般就是模型的参数。当模型复杂的时候共享变量会无比复杂。官网给了一个case，当创建两层卷积的过滤器时，每输入一次图片就会创建一次过滤器对应的变...

宜配屋