PyTorch使用cpu加载模型运算方式

yipeiwu_com5年前 (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设计】。

返回列表

不管是用import还是用from mmmm import *的方式导入模块，当程序运行之后，回头在看那个存储着mmmm.py文件的目录中，多了一个文件: $ ls mmm* mmm...

我们首先来看下全部代码： # -*- coding: cp936 -*- import win32serviceutil import win32service import...

本文实例讲述了Python3遍历目录树的方法。分享给大家供大家参考。具体实现方法如下： import os, fnmatch # 检查一个目录，后者某个包含子目录的目录树，并根据某种...

小编今天教你们python怎么导入坐标点，解决你在生活中遇到的小问题。首先下载安装python，打开文本编辑器，将文件保存成 py格式，如果python目录不在usr/bin目录下，则...

前言 Matplotlib 是 Python 的绘图库。它可与 NumPy 一起使用，提供了一种有效的 MatLab 开源替代方案。它也可以和图形工具包一起使用，如 PyQt 和 w...

宜配屋