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

更新时间：2020年01月13日 09:51:02 作者：ngy321

今天小编就为大家分享一篇PyTorch使用cpu加载模型运算方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

没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加载模型运算方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python3爬虫ChromeDriver的安装实例
在本篇文章里小编给大家整理的是一篇关于Python3爬虫ChromeDriver的安装实例内容，有兴趣的朋友们可以学习参考下。
2021-02-02
Python中进行表单处理的方法详解
这篇文章主要为大家详细介绍了Python中进行表单处理的方法,分别是利用Flask和Flask-WTF处理表单,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下
2023-12-12
python缩进长度是否统一
在本篇内容里小编给大家整理的是一篇关于python缩进长度是否统一的相关知识点，需要的朋友们可以学习下。
2020-08-08
Python BeautifulSoup基本用法详解(通过标签及class定位元素)
这篇文章主要介绍了Python BeautifulSoup基本用法(通过标签及class定位元素),本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-08-08
Python count函数使用方法实例解析
这篇文章主要介绍了Python count函数使用方法实例解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-03-03
Python读取xlsx文件的实现方法
这篇文章主要介绍了Python读取xlsx文件的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-07-07
python代码如何实现切换中英文输入法
这篇文章主要介绍了python代码如何实现切换中英文输入法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-11-11
详解Python如何巧妙实现数学阶乘n!
一个正整数的阶乘（factorial）是所有小于及等于该数的正整数的积，并且0的阶乘为1。自然数n的阶乘写作n!，本文就给大家介绍如何使用python和第三方库来实现数学运算中的阶乘以及阶乘累计求和
2023-03-03
python 远程统计文件代码分享
享一个Python获取远程文件大小的函数代码，简单实用，是学习Python编程的基础实例。
2015-05-05
跟老齐学Python之让人欢喜让人忧的迭代
跟一些比较牛X的程序员交流，经常听到他们嘴里冒出一个不标准的英文单词，而loop、iterate、traversal和recursion如果不在其内，总觉得他还不够牛X。当让，真正牛X的绝对不会这么说的，他们只是说“循环、迭代、遍历、递归”，然后再问“这个你懂吗？”。
2014-10-10