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

返回列表

如何在win7+Python3.5的环境下安装成功scrapy？通过pip3 install Scrapy直接安装，一般会报错：error: Unable to find vcvars...

不久前，在互联网上出现了一篇有趣的文章，讲的是对于同一个问题，不同层次的Python程序员编出的Python代码，显示出了不同的风格，代码都很简单，有趣。下面让我们一起来看看一个Pyth...

Python3 abs() 函数描述 abs() 函数返回数字的绝对值。语法以下是 abs() 方法的语法: abs( x ) 参数 x-- 数值表达式，可以是整数，浮点...

代码代码很简单，主要是为了熟悉Selenium这个库的函数，为后续的短信轰炸做个铺垫 from selenium import webdriver import time imp...

如题，解决Python中用PyQt时中文乱码问题的解决方法：在中文字符串前面加上u，如u'你好，世界'，其他网上的方法没有多去探究，Python的版本也会影响解决方法，故这里只推荐这种...

宜配屋