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

返回列表

python里面可以将路径里面的\替换成/避免转义。 os.walk方法可以将目标路径下文件的root，dirs，files提取出来。后面对每个文件进行操作。切片操作[:]判断是否为....

本文实例讲述了Python3.6连接Oracle数据库的方法。分享给大家供大家参考，具体如下：下载cx_Oracle模块模块： https://pypi.python.org/pypi...

本文实例讲述了Python实现对文件进行单词划分并去重排序操作。分享给大家供大家参考，具体如下：文件名：test1.txt 文件内容： But soft what light thr...

本文实例讲述了Python实现彩票机选器的方法。分享给大家供大家参考。具体实现方法如下： # -*- coding: utf8 -*- from Tkinter import * i...

1、下载pycharm pycharm是一种Python IDE，能够帮助我们在编写代码时提高效率。网上提供的有专业版和教育版之分。专业版是收费的，功能更全面点。教育版...

宜配屋