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

返回列表

如下所示： # -*- coding:utf-8 -*- import random,urllib,urllib2 import re,time x=input("请输入需要注册...

如下所示： import os DIR = "/home/serho/workspace/lisp" def compare(x, y): stat_x = os.stat(D...

manage.py是在我们创建Django项目的时候就自动生成在根目录下的一个命令行工具，它可以执行一些简单的命令，其功能是将Django project放到sys.path目录中，同时...

这篇文章主要介绍了python文字转语音实现过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下使用百度接口接口地址 ...

本文以实例形式分析了Python多进程编程技术，有助于进一步Python程序设计技巧。分享给大家供大家参考。具体分析如下：一般来说，由于Python的线程有些限制，例如多线程不能充分利...

宜配屋