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

返回列表

Elasticsearch是一个分布式、Restful的搜索及分析服务器，Apache Solr一样，它也是基于Lucence的索引服务器，但我认为Elasticsearch对比Solr...

如下所示： import tkinter win=tkinter.Tk() text=tkinter.Text(win) #文本编辑器(用于展示数据) text.insert(t...

1. File->Setting 打开如下所示的对话框，可以在Editor->General->Font下设置字体大小（Size）、行距（Line spac...

如下所示： # -*- coding: utf-8 -*- import sys import os from PyQt5 import QtCore, QtGui, QtWidge...

复制代码代码如下:import socket#socket通信客户端def client(): mysocket=socket.socket(soc...

宜配屋