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

返回列表

1. 背景在深度学习的任务中，通常需要比较长时间的训练，因此我们会选择离开电脑。笔者在跟踪模型表现，观察模型accuracy 以及 loss 的时候，比较传统的方法是在控制台prin...

最近有一个学弟问我一些求解质因数的问题，帮他解决问题的同时自己也试着写了几个差不多效果的脚本，有很多不同的思路，以下是相关脚本。 n = int(input("input numbe...

出现原因：缺失相应的whl文件。解决办法：下载并安装对应的whl文件。提供一个whl文件的下载网址：http://www.lfd.uci.edu/~gohlke/pythonlibs...

为什么要实现分页？在大部分网站中分页的功能都是必要的，尤其是在后台管理中分页更是不可或缺分页能带给用户更好的体验，也能减轻服务器的压力对于分页来说，有许多方法都可以实现例如把数据...

以下代码以Python3.6.1 / windows10为例 Less is more! #!/usr/bin/env python # coding=utf-8 __author...

宜配屋