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实现收支的自动计算，能够查询每笔账款的消费详情，具体内容如下 1、函数需要两个文件：一个类似钱包功能，存放钱；另一个用于记录每笔花销的用途 #!/usr/bin/env...

QCalendarWidget 是日历控件。它允许用户以简单和直观的方式选择日期。 #!/usr/bin/python3 # -*- coding: utf-8 -*- """ P...

概述如果程序处理的数据比较多、比较复杂，那么在程序运行的时候，会占用大量的内存，当内存占用到达一定的数值，程序就有可能被操作系统终止，特别是在限制程序所使用的内存大小的场景，更容易发...

具体内容如下：使用sorted将字典按照其value大小排序 >>> record = {'a':89, 'b':86, 'c':99, 'd':100} &g...

宜配屋