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

返回列表

1、下载安装好PyCharm 专业版后打开或者新建一个Python项目，找到View导航栏，如下图：在Tool Windows下可以找到SciView按钮，但是每次打开PyChar...

本篇文章对隐马尔可夫模型的前向和后向算法进行了Python实现，并且每种算法都给出了循环和递归两种方式的实现。前向算法Python实现循环方式 import numpy as...

pandas进行打印，控制台的显示默认是换行的在pycharm中的控制台也没有办法设置可加入如下代码打印，结果不换行，看着数据更加直观 import pandas as pd i...

本文实例总结了Python操作redis方法。分享给大家供大家参考，具体如下： python连接方式可参考：/post/161353.htm 这里介绍详细使用 1、String 操作 r...

循环写入字典key、value、删除指定的键值对：原文本‘jp_url.txt'每行元素以逗号分隔： host_key,product_id,product_name,cont_s...

宜配屋