PyTorch 随机数生成占用 CPU 过高的解决方法


Posted in Python onJanuary 13, 2020

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 相关文章推荐
python类继承与子类实例初始化用法分析
Apr 17 Python
python通过百度地图API获取某地址的经纬度详解
Jan 28 Python
使用pandas中的DataFrame数据绘制柱状图的方法
Apr 10 Python
python爬取足球直播吧五大联赛积分榜
Jun 13 Python
python+selenium实现自动抢票功能实例代码
Nov 23 Python
Python 实现自动完成A4标签排版打印功能
Apr 09 Python
解决Keras TensorFlow 混编中 trainable=False设置无效问题
Jun 28 Python
解决TensorFlow调用Keras库函数存在的问题
Jul 06 Python
Python 解析库json及jsonpath pickle的实现
Aug 17 Python
Python实现小黑屋游戏的完整实例
Jan 06 Python
pytorch中F.avg_pool1d()和F.avg_pool2d()的使用操作
May 22 Python
python中mongodb包操作数据库
Apr 19 Python
python批量处理txt文件的实例代码
Jan 13 #Python
Python hashlib常见摘要算法详解
Jan 13 #Python
Pytorch释放显存占用方式
Jan 13 #Python
pytorch 实现删除tensor中的指定行列
Jan 13 #Python
python基于event实现线程间通信控制
Jan 13 #Python
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
Jan 13 #Python
python、PyTorch图像读取与numpy转换实例
Jan 13 #Python
You might like
萌王史莱姆”萌王性别尴尬!那“萌战”归女组还是男?
2018/12/17 日漫
让的PHP代码飞起来的40条小技巧(提升php效率)
2010/04/12 PHP
php利用新浪接口查询ip获取地理位置示例
2014/01/20 PHP
PHP学习笔记之字符串编码的转换和判断
2014/05/22 PHP
php网站被挂木马后的修复方法总结
2014/11/06 PHP
PHP实现时间比较和时间差计算的方法示例
2017/07/24 PHP
JavaScript 计算图片加载数量的代码
2011/01/01 Javascript
jquery图片上下tab切换效果
2011/03/18 Javascript
js鼠标点击事件在各个浏览器中的写法及Event对象属性介绍
2013/01/24 Javascript
javascript中的__defineGetter__和__defineSetter__介绍
2014/08/15 Javascript
Node.js中使用计时器定时执行函数详解
2014/08/15 Javascript
jQuery Mobile操作HTML5的常用函数总结
2016/05/17 Javascript
jQuery选择器实例应用
2017/01/05 Javascript
jQuery学习之DOM节点的插入方法总结
2017/01/22 Javascript
js 数据存储和DOM编程
2017/02/09 Javascript
Node.JS更改Windows注册表Regedit的方法小结
2017/08/18 Javascript
vue项目部署上线遇到的问题及解决方法
2018/06/10 Javascript
vue-cli3.0如何使用CDN区分开发、生产、预发布环境
2018/11/22 Javascript
详解vue中使用vue-quill-editor富文本小结(图片上传)
2019/04/24 Javascript
小程序封装路由文件和路由方法(5种全解析)
2019/05/26 Javascript
react 组件传值的三种方法
2019/06/03 Javascript
jQuery实现鼠标放置名字上显示详细内容气泡提示框效果的方法分析
2020/04/04 jQuery
Python实现Linux的find命令实例分享
2017/06/04 Python
使用python Fabric动态修改远程机器hosts的方法
2018/10/26 Python
Python实现的爬取豆瓣电影信息功能案例
2019/09/15 Python
Python3.9 beta2版本发布了,看看这7个新的PEP都是什么
2020/06/10 Python
香奈儿美国官网:CHANEL美国
2020/05/20 全球购物
承诺书怎么写
2014/03/26 职场文书
团拜会策划方案
2014/06/07 职场文书
精神文明建设标语
2014/06/16 职场文书
篮球赛口号
2014/06/18 职场文书
小学生民族团结演讲稿
2014/08/27 职场文书
2015年党风建设工作总结
2015/04/29 职场文书
使用springboot暴露oracle数据接口的问题
2021/05/07 Oracle
MySQL索引是啥?不懂就问
2021/07/21 MySQL
动视暴雪取消疫苗禁令 让所有员工返回线下工作
2022/04/03 其他游戏