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.新建一个项目新建一个maven，并且选择webapp类型。 2.点击next选项这里面的两个选项可以随便填，但是Artif...

利用pymongo包进行数据库的连接，使用xlrd包读取excel数据，由于二者数据结构的不同，要将excel格式数据转换为json格式数据。由于编码问题会出现“TypeError: '...

1. 使用.logfile 方法 #!/usr/bin/env python import pexpect import sys host="146.11.85.xxx" user=...

如下所示： writer = csv.writer(open('xx.csv','w')) data = [('zzz', 20, 123456),('jojo', 10, 7890...

一、python3下面的pyttsx3正常可以使用，类似下面： import pyttsx3 engine = pyttsx3.init() rate = engine.getPro...

宜配屋