Pytorch CPU运行与GPU运行

75 阅读 0 评论 50 点赞

我是靠谱客的博主留胡子花瓣，最近开发中收集的这篇文章主要介绍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 CPU运行与GPU运行所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

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