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

返回列表

视频对象提取与其说是视频对象提取，不如说是视频颜色提取，因为其本质还是使用了OpenCV的HSV颜色物体检测。下面话不多说了，来一起看看详细的介绍吧。 HSV介绍 HSV分别代表，...

暂时是一个手动设置无向图中的边，用一个二维数组表示，后面会改进为用户自己定义无向图的边。学习python的新手，若大佬有解决的办法，希望不吝赐教 #无向图判断环是否存在 def d...

一、推荐安装Anaconda 官方介绍：Anaconda is a completely free Python distribution (including for commerci...

本文实例讲述了Python变量、数据类型、数据类型转换相关函数用法。分享给大家供大家参考，具体如下： python变量的使用不需要进行类型声明（类型名变量名），给一个变量名赋什么值就是...

本文分享的实例主要是Python编程二分法实现冒泡算法+快速排序，具体如下。冒泡算法： #-*- coding: UTF-8 -*- #冒泡排序 def func(lt): if...

宜配屋