首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 Python

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 过高的解决方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

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

- Author -

lucasgyshen

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐
python里将list中元素依次向前移动一位
Sep 12 Python
简单讲解Python编程中namedtuple类的用法
Jun 21 Python
python获取多线程及子线程的返回值
Nov 15 Python
python爬取网页内容转换为PDF文件
Jul 28 Python
python实现可视化动态CPU性能监控
Jun 21 Python
Python + OpenCV 实现LBP特征提取的示例代码
Jul 11 Python
Python自省及反射原理实例详解
Jul 06 Python
python如何实现图片压缩
Sep 11 Python
python基于Kivy写一个图形桌面时钟程序
Jan 28 Python
Python页面加载的等待方式总结
Feb 28 Python
Python3 如何开启自带http服务
May 18 Python
Python中生成随机数据安全性、多功能性、用途和速度方面进行比较
Apr 14 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
Android Studio(4) seata(1) cv2(1) 迪杰斯特拉(1) MyBatis(5) nacos(1) 优先队列(1) 三角函数(1) exec(1) Dijkstra(1)
You might like
php mssql 时间格式问题
2009/01/13 PHP
Linux下php5.4启动脚本
2014/08/03 PHP
Symfony页面的基本创建实例详解
2015/01/26 PHP
php使用wordwrap格式化文本段落的方法
2015/03/17 PHP
PHP 获取 ping 时间的实现方法
2017/09/29 PHP
window.print打印指定div实例代码
2013/12/13 Javascript
JS中Date日期函数中的参数使用介绍
2014/01/02 Javascript
每天一篇javascript学习小结（属性定义方法）
2015/11/19 Javascript
Underscore源码分析
2015/12/30 Javascript
利用JavaScript阻止表单提交的两种方法
2016/08/11 Javascript
ES6中的数组扩展方法
2016/08/26 Javascript
JQuery遍历元素的后代和同胞实现方法
2016/09/18 Javascript
深入理解JS中的Function.prototype.bind()方法
2016/10/11 Javascript
JavaScript注入漏洞的原理及防范(详解)
2016/12/04 Javascript
JavaScript实现水平进度条拖拽效果
2017/01/18 Javascript
Angular数据绑定机制原理
2018/04/17 Javascript
解决三元运算符 报错“SyntaxError: can''t assign to conditional expression”
2020/02/12 Javascript
jQuery实现html可联动的百分比进度条
2020/03/26 jQuery
js实现登录时记住密码的方法分析
2020/04/05 Javascript
解决vue axios跨域 Request Method: OPTIONS问题(预检请求)
2020/08/14 Javascript
[55:16]Mski vs VGJ.S Supermajor小组赛C组 BO3 第二场 6.3
2018/06/04 DOTA
[52:02]DOTA2-DPC中国联赛 正赛 Phoenix vs Dragon BO3 第二场 2月26日
2021/03/11 DOTA
在Python中用has_key()方法查找键是否存在的教程
2015/05/21 Python
python算法演练_One Rule 算法(详解)
2017/05/17 Python
详解django中使用定时任务的方法
2018/09/27 Python
Python实现的删除重复文件或图片功能示例【去重】
2019/04/23 Python
python topk()函数求最大和最小值实例
2020/04/02 Python
Book Depository澳大利亚：世界领先的专业在线书店之一
2018/12/27 全球购物
工商管理专业学生的自我评价
2013/10/01 职场文书
护理专业个人求职简历的自我评价
2013/10/13 职场文书
博士研究生自我鉴定范文
2013/12/04 职场文书
大学生党员承诺书
2014/05/20 职场文书
出差报告怎么写
2014/11/06 职场文书
python爬虫之利用selenium模块自动登录CSDN
2021/04/22 Python
基于Redis实现分布式锁的方法（lua脚本版）
2021/05/12 Redis
pytorch Dropout过拟合的操作
2021/05/27 Python