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是当今日趋流行的一种脚本语言，它比Java更简单，比php更强大，并且还适用于做桌面应用的开发，在ubuntu中，更加是必须的一种脚本引擎，所以有必要去学习一下，这里的文章只...

代码：复制代码代码如下: import os while True: dynamic = input('输入计算表达式：') if dynamic != 'cls': try: re...

本文实例讲述了Python基于递归和非递归算法求两个数最大公约数、最小公倍数。分享给大家供大家参考，具体如下：最大公约数和最小公倍数的概念大家都很熟悉了，在这里就不多说了，今天这个是因...

*****看一下我定义的change（）和run（）函数****** 绘图坐标体系：作用：设置主窗体的大小和位置 turtle.setup(width, height, sta...

python函数修饰符@的作用是为现有函数增加额外的功能，常用于插入日志、性能测试、事务处理等等。创建函数修饰符的规则：（1）修饰符是一个函数（2）修饰符取被修饰函数为参数...

宜配屋