1. 首页
  2. 文章中心
  3. 深度学习

Pytorch CPU运行与GPU运行

198 阅读 0 评论
我是靠谱客的博主 留胡子花瓣，这篇文章主要介绍Pytorch CPU运行与GPU运行，现在分享给大家，希望可以做个参考。

需要在Pytorch下跑一SalGAN，但是代码居然用CPU来计算，将其改为GPU版。Pytorch下使用GPU计算需要三步,记录一下：

1.将数据载入GPU：

input -> input.cuda()

2.将模型载入GPU

model = model.cuda()

3.将输出的预测转为numpy格式

result.data.numpy() -> result.data.cpu().numpy()

 

最后

以上就是留胡子花瓣最近收集整理的关于Pytorch CPU运行与GPU运行的全部内容，更多相关Pytorch内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
点赞(131)

相关文章

论文笔记 —— 《Transfer Feature Learning with Joint Distribution Adaptation》
论文笔记 —— 《Transfer Feature Learning with Joint Distribution Adaptation》
[2022]李宏毅深度学习与机器学习第十一讲（必修）-Domain Adaptation做笔记的目的Domain ShiftDomain Adaptation
[2022]李宏毅深度学习与机器学习第十一讲（必修）-Domain Adaptation做笔记的目的Domain ShiftDomain Adaptation
最优传输论文（六）：Wasserstein Distance Guided Representation Learning for Domain Adaptation论文原理前言原理阐述
最优传输论文（六）：Wasserstein Distance Guided Representation Learning for Domain Adaptation论文原理前言原理阐述
对齐与分布图层
对齐与分布图层
Pytorch CPU运行与GPU运行
Pytorch CPU运行与GPU运行
Contextual Encoder-Decoder Network for Visual Saliency Prediction【论文笔记】
Contextual Encoder-Decoder Network for Visual Saliency Prediction【论文笔记】
Adversarial Nets Papers AdversarialNetsPapers The classical Papers about adversarial nets The First paper  [Generative Adversarial Nets] [Paper] [Code](the first paper abou
Adversarial Nets Papers AdversarialNetsPapers The classical Papers about adversarial nets The First paper  [Generative Adversarial Nets] [Paper] [Code](the first paper abou
Visual Saliency Prediction with Generative Adversarial Networks
Visual Saliency Prediction with Generative Adversarial Networks

评论列表共有 0 条评论

立即
投稿
返回
顶部