编程 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进阶教程之循环相关函数range、enumerate、zip

Aug 30 Python

Django中对通过测试的用户进行限制访问的方法

Jul 23 Python

Python中文竖排显示的方法

Jul 28 Python

使用python绘制常用的图表

Aug 27 Python

Python subprocess模块详细解读

Jan 29 Python

python3.6+opencv3.4实现鼠标交互查看图片像素

Feb 26 Python

Python文件监听工具pyinotify与watchdog实例

Oct 15 Python

Python 在OpenCV里实现仿射变换—坐标变换效果

Aug 30 Python

python根据文本生成词云图代码实例

Nov 15 Python

Python3.5 win10环境下导入kera/tensorflow报错的解决方法

Dec 19 Python

Python模块/包/库安装的六种方法及区别

Feb 24 Python

python读取excel进行遍历/xlrd模块操作

Jul 12 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采集类Snoopy抓取图片实例

2014/06/19 PHP

PHP常用数组函数介绍

2014/07/28 PHP

FireFox与IE 下js兼容触发click事件的代码

2008/11/20 Javascript

jQuery事件 delegate()使用方法介绍

2012/10/30 Javascript

js文件Cookie存取值示例代码

2014/02/20 Javascript

怎么判断js脚本加载完成

2014/02/28 Javascript

js文件包含的几种方式介绍

2014/09/28 Javascript

js ajaxfileupload.js上传报错的解决方法

2016/05/05 Javascript

JavaScript中有关一个数组中最大值和最小值及它们的下表的输出的解决办法

2016/07/01 Javascript

使用JS组件实现带ToolTip验证框的实例代码

2017/08/23 Javascript

D3.js实现简洁实用的动态仪表盘的示例

2018/04/04 Javascript

vue-cli3项目展示本地Markdown文件的方法

2019/06/07 Javascript

微信小程序中悬浮窗功能的实现代码

2019/08/02 Javascript

JS防抖和节流实例解析

2019/09/24 Javascript

vue父子组件间引用之$parent、$children

2020/05/20 Javascript

通过实例浅析Python对比C语言的编程思想差异

2015/08/30 Python

Python 递归函数详解及实例

2016/12/27 Python

python中的colorlog库使用详解

2019/07/05 Python

Pyspark读取parquet数据过程解析

2020/03/27 Python

tensorflow与numpy的版本兼容性问题的解决

2021/01/08 Python

经济实惠的豪华家具：My-Furniture

2019/03/12 全球购物

Ruby如何实现动态方法调用

2012/11/18 面试题

高中生校园生活自我评价

2013/09/19 职场文书

饭店工作计划书

2014/01/10 职场文书

工程招投标邀请书

2014/01/30 职场文书

关于是否需要写商业计划书

2014/02/07 职场文书

爱国主义演讲稿

2014/05/07 职场文书

优秀团员事迹材料

2014/12/25 职场文书

异地恋情人节寄语

2015/02/28 职场文书

2015年团队工作总结范文

2015/05/04 职场文书

电影地道战观后感

2015/06/04 职场文书

篮球比赛通讯稿

2015/07/18 职场文书

《灰雀》教学反思

2016/02/19 职场文书

《秋思》教学反思

2016/02/23 职场文书

Python内置数据结构列表与元组示例详解

2021/08/04 Python

阿里云服务器部署mongodb的详细过程

2021/09/04 MongoDB