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

返回列表

python 绘制拟合曲线并加指定点标识 import os import numpy as np from scipy import log from scipy.optimiz...

使用python + shell 编写，是一个简易solaris系统巡检程序复制代码代码如下:#!/usr/bin/python -u#-*- coding:utf-8 -*-''...

自定义查询对象 - objects ①声明一个类EntryManager，继承自models.Manager，并添加自定义函数 ②使用创建的自定义类EntryManager 覆盖Mo...

今天因工作需要写了个小程序，用于在图片集中自动抽取需要的照片。该程序只是实现了基本功能，还有很多需要完善的地方，展示出来算是给自己鼓鼓气吧。该程序应用有一定特殊条件，因我选择的图片集是...

本文实例讲述了Python实现的摇骰子猜大小功能小游戏。分享给大家供大家参考，具体如下：最近学习Python的随机数，逻辑判断，循环的用法，就想找一些练习题，比如小游戏猜大小，程序思路...

宜配屋