pytorch在CPU和GPU上加载模型

pytorch允许把在GPU上训练的模型加载到CPU上，也允许把在CPU上训练的模型加载到GPU上。

CPU->CPU,GPU->GPU

torch.load('gen_500000.pkl')

GPU->CPU

torch.load('gen_500000.pkl', map_location=lambda storage, loc: storage)

CPU->GPU1

torch.load('gen_500000.pkl', map_location=lambda storage, loc: storage.cuda(1))

相关文章：

2022-12-23
2022-12-23
2022-12-23
2022-12-23
2021-04-01
2022-12-23
2022-12-23
2022-01-29

猜你喜欢

2021-09-28
2022-12-23
2022-12-23
2022-01-01
2021-08-02
2022-12-23
2022-12-23

相关资源

下载 2023-01-26
下载 2022-12-01
下载 2021-06-05

相似解决方案

热门标签

Java Python linux javascript Mysql C# Docker 算法前端 SpringBoot Redis Vue spring 设计模式 .net core .net kubernetes c++ 数据库数据结构大数据 js 机器学习微服务 Android Go 程序员面试 JVM ASP.net core 云原生人工智能后端 PHP git CSS golang k8s Nginx Django mybatis 深度学习多线程 React 架构 devops 爬虫云计算 Spring Boot LeetCode