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实现DNS正向查询、反向查询的例子
Apr 25 Python
Tensorflow卷积神经网络实例进阶
May 24 Python
Python面向对象之类和对象实例详解
Dec 10 Python
python pygame实现方向键控制小球
May 17 Python
Python实现微信机器人的方法
Sep 06 Python
python实现代码统计器
Sep 19 Python
wxPython绘图模块wxPyPlot实现数据可视化
Nov 19 Python
Python实现名片管理系统
Feb 14 Python
基于python 取余问题(%)详解
Jun 03 Python
解决Keras 自定义层时遇到版本的问题
Jun 16 Python
如何基于matlab相机标定导出xml文件
Nov 02 Python
pycharm 如何取消连按两下shift出现的全局搜索
Jan 15 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
PHP中include()与require()的区别说明
2017/02/14 PHP
Nigma vs Alliance BO5 第二场2.14
2021/03/10 DOTA
Nigma vs Alliance BO5 第四场2.14
2021/03/10 DOTA
javascript下阻止表单重复提交、防刷新、防后退
2007/08/17 Javascript
Js切换功能的简单方法
2010/11/23 Javascript
Prototype源码浅析 String部分(二)
2012/01/16 Javascript
Javascript图像处理—亮度对比度应用案例
2013/01/03 Javascript
js获取下拉列表的值和元素个数示例
2014/05/07 Javascript
JavaScript中匿名、命名函数的性能测试
2014/09/04 Javascript
js获取页面传来参数的方法
2014/09/06 Javascript
JS实现点击文字对应DIV层不停闪动效果的方法
2015/03/02 Javascript
JavaScript中的toLocaleDateString()方法使用简介
2015/06/12 Javascript
javascript手风琴下拉菜单实现代码
2015/11/12 Javascript
JS清除字符串中重复值的实现方法
2016/08/03 Javascript
AngularJS 单元测试(一)详解
2016/09/21 Javascript
vue实现列表的添加点击
2016/12/29 Javascript
js选项卡的制作方法
2017/01/23 Javascript
Javascript下拉刷新的简单实现
2017/02/14 Javascript
详解angular2 控制视图的封装模式
2018/12/27 Javascript
Angular脚手架开发的实现步骤
2019/04/09 Javascript
微信小程序 wxParse插件显示视频问题
2019/09/27 Javascript
使用Python对SQLite数据库操作
2017/04/06 Python
bluepy 一款python封装的BLE利器简单介绍
2019/06/25 Python
python 随机生成10位数密码的实现代码
2019/06/27 Python
django框架用户权限中的session缓存到redis中的方法
2019/08/06 Python
安装Pycharm2019以及配置anconda教程的方法步骤
2019/11/11 Python
Python实现word2Vec model过程解析
2019/12/16 Python
Python嵌入C/C++进行开发详解
2020/06/09 Python
一款利用纯css3实现的win8加载动画的实例分析
2014/12/11 HTML / CSS
英国男女奢华内衣和泳装购物网站:Figleaves
2017/01/28 全球购物
乌克兰设计师和品牌的服装:Love&Live
2020/04/14 全球购物
正规的求职信范文分享
2013/12/11 职场文书
酒店七夕情人节活动策划方案
2014/08/24 职场文书
2014年团员学习十八大思想汇报
2014/09/13 职场文书
工作经验交流材料
2014/12/30 职场文书
css3 选择器
2022/05/11 HTML / CSS