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

返回列表

一、简单使用复制代码代码如下: def TestLogBasic(): import logging l...

实例如下所示： #!/usr/bin/python # -*- coding:utf8 -*- from selenium import webdriver import os d...

示例效果颜色线的风格标记类型 plot的更多参数以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持【听图阁-专注于Python设计】。...

在Linux中，可以使用nohup将脚本放置后台运行，如下： nohup python myscript.py params1 > nohup.out 2>&1 &...

如下所示： import json f = open("index.html", "wb") json.dump(response.data, f) f.close() dum...

宜配屋