PyTorch 随机数生成占用 CPU 过高的解决方法
更新时间：2020年01月13日 09:40:52 作者：lucasgyshen
今天小编就为大家分享一篇PyTorch 随机数生成占用 CPU 过高的解决方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
PyTorch 随机数生成占用 CPU 过高的问题
今天在使用 pytorch 的过程中，发现 CPU 占用率过高。经过检查，发现是因为先在 CPU 中生成了随机数，然后再调用.to(device)传到 GPU，这样导致效率变得很低，并且CPU 和 GPU 都被消耗。
查阅PyTorch文档后发现，torch.randn(shape, out)可以直接在GPU中生成随机数，只要shape是tensor.cuda.Tensor类型即可。这样，就可以避免在 CPU 中生成过大的矩阵，而 shape 变量是很小的。
因此，下面的代码就可以进行这种操作了。
noise = torch.cuda.FloatTensor(shape) if torch.cuda.is_available() else torch.FloatTensor(shape) torch.randn(shape, out=noise)
以上这篇PyTorch 随机数生成占用 CPU 过高的解决方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python中对字典的几个处理方法分享，文章围绕主题展开详细的内容介绍，具有一定的参考价值，感兴趣的小伙伴可以参考一下2022-08-08
这篇文章主要为大家详细介绍了python实现宿舍管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-11-11
在本篇文章里小编给大家整理的是关于python实发邮件的相关知识点内容，有需要的朋友们学习下。2019-11-11
今天小编就为大家分享一篇python数据批量写入ScrolledText的优化方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
这篇文章主要介绍了python获取文件版本信息、公司名和产品名的方法,是Python程序设计中非常实用的技巧,需要的朋友可以参考下2014-10-10
本文主要介绍了pandas按某列降序的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-01-01
这篇文章主要介绍了python Yaml 、Json 、Dict 之间的转化的示例，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下2020-10-10
这篇文章主要介绍了Python3 读、写Excel文件的操作方法,需要的朋友可以参考下2018-10-10
这篇文章主要介绍了python将下载到本地m3u8视频合成MP4的代码详解,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
这篇文章主要介绍了python格式化输出保留2位小数的实现方法,需要的朋友可以参考下2019-07-07
最新评论