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

返回列表

本文在前面文章基础上介绍tkinter添加图片和文本，在这之前，我们需要安装一个图片库，叫Pillow，这个需要下载exe文件，根据下面图片下载和安装。下载完后直接双击安装exe，默...

注：本文针对单个服务器上多块GPU的使用，不是多服务器多GPU的使用。在一些实验中，由于Batch_size的限制或者希望提高训练速度等原因，我们需要使用多块GPU。本文针对Pytor...

这个程序的功能非常的简单，就是每天在系统中新建一个文件夹。文件夹即当前的时间。此代码是在同事那边看到的，为了锻炼下自己薄弱的Python能力，所以花时间重新写了一个。具体代码如下：...

在使用python做爬虫的过程中，有些页面的的部分数据是通过js异步加载的，js调用接口的请求中有时还带有些加密的参数很难破解无法使用requests这样的包直接爬取数据，因此需要借助s...

用matplotlib.pyplot画的图，显示和保存的图片周围都会有白边，可以去掉。为了显示的更清楚，给图片加了红色的框代码 “` import matplotlib.pypl...

宜配屋