2020-01-15 将数据从CPU赋值到GPU变量上

这个是在最近看代码中看到的问题, 有时候我们在GPU上已经定义好了变量,但是需要的数据却是在CPU上.

那么这个时候我们可能会用 resize_as_ + copy 的方式来解决这个问题.

  for i, data in enumerate(train_loader):
        real_cpu, _ = data
        input = torch.FloatTensor(64, 3, 64, 64)
        input.resize_as_(real_cpu).copy_(real_cpu)

在这里可以这么用,这段代码出自D2GAN, 感兴趣的朋友可以看看.

最后,祝大家新年快乐!!!!

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • iOS性能优化(1)-CPU和GPU
    性能优化是一个很重要的一部分,我们首先看CPU和GPU的部分,想知道CPU和GPU是怎么优化的,就必须要明白CPU...
    周灬阅读 2,287评论 1赞 7
  • 浅析GPU计算——CPU和GPU的选择
    原文地址：https://blog.csdn.net/breaksoftware/article/details/...
    笑火云天阅读 601评论 0赞 2
  • faster rcnn论文笔记和代码分析
    目录faster rcnn论文备注caffe代码框架简介faster rcnn代码分析后记 faster rcnn...
    db24cc阅读 9,648评论 2赞 12
  • GPU vs CPU in iOS
    一直以来，我们做产品的时候并没有特别的去考虑CPU/GPU的使用，最近为了提升可视化功能的性能，发现合理使用GPU...
    雨_树阅读 3,584评论 0赞 9
  • 从GPU到GPGPU（转）
    GPU的工作原理GPU的起源可以追溯到个人电脑开始普及的80年代，此时用户开始有了游戏的需求。为了更好的用户体验，...
    听风阁阅读 6,432评论 0赞 7