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

返回列表

实现对图像进行简单的高斯去噪和椒盐去噪。代码如下： import numpy as np from PIL import Image import matplotlib.pyplo...

本文实例讲述了Python反射用法。分享给大家供大家参考，具体如下： class Person: def __init__(self): self.name = "zjg...

起因说起来录制视频，我们可能有很多的软件，但是比较坑的是，好像很少的软件支持能够同时录制两个摄像头的视频，于是我们用python自己写一个。要是OpenCV+python。貌似很简单就...

Python元字典字典(dictionary)是除列表以外python之中最灵活的内置数据结构类型。列表是有序的对象结合，字典是无序的对象集合。两者之间的区别在于：字典当中的元素是通...

本文实例讲述了python检测远程udp端口是否打开的方法。分享给大家供大家参考。具体实现方法如下：复制代码代码如下:import socket import threading i...

宜配屋