编程 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中使用md5sum检查目录中相同文件代码分享

Feb 02 Python

使用Python的Zato发送AMQP消息的教程

Apr 16 Python

浅析Git版本控制器使用

Dec 10 Python

Python cookbook（数据结构与算法）保存最后N个元素的方法

Feb 13 Python

Sanic框架路由用法实例分析

Jul 16 Python

python中字符串内置函数的用法总结

Sep 13 Python

kafka-python 获取topic lag值方式

Dec 23 Python

Django REST Swagger实现指定api参数

Jul 07 Python

详解Python调用系统命令的六种方法

Jan 28 Python

pytorch下的unsqueeze和squeeze的用法说明

Feb 06 Python

python中的plt.cm.Paired用法说明

May 31 Python

python游戏开发之pygame实现接球小游戏

Apr 22 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

详解：――如何将图片储存在数据库里

2006/12/05 PHP

php流量统计功能的实现代码

2012/09/29 PHP

7个鲜为人知却非常实用的PHP函数

2015/07/01 PHP

WordPress中用于获取文章信息以及分类链接的函数用法

2015/12/18 PHP

PHP自带方法验证邮箱、URL、IP是否合法的函数

2016/12/08 PHP

php创建图像具体步骤

2017/03/13 PHP

Mootools 1.2教程(2) DOM选择器

2009/09/14 Javascript

jquery 图片上一张下一张链接效果（续篇）

2010/04/20 Javascript

jQuery源码分析-01总体架构分析

2011/11/14 Javascript

js使用函数绑定技术改变事件处理程序的作用域

2011/12/26 Javascript

jquery offset函数应用实例

2012/11/14 Javascript

node.js中的fs.mkdirSync方法使用说明

2014/12/17 Javascript

javascript限制文本框输入值类型的方法

2015/05/07 Javascript

JS实现的仿东京商城菜单、仿Win右键菜单及仿淘宝TAB特效合集

2015/09/28 Javascript

基于BootStrap Metronic开发框架经验小结【七】数据的导入、导出及附件的查看处理

2016/05/12 Javascript

window.open不被拦截的简单实现代码(推荐)

2016/08/04 Javascript

简单实现jQuery弹窗效果

2017/10/30 jQuery

解决layui使用layui-icon出现默认图标的问题

2019/09/11 Javascript

JS常用正则表达式超全集(密码强度校验,金额校验,IE版本,IPv4,IPv6校验)

2020/02/03 Javascript

Python复制目录结构脚本代码分享

2015/03/06 Python

Python模块WSGI使用详解

2018/02/02 Python

python PyTorch参数初始化和Finetune

2018/02/11 Python

对Python中内置异常层次结构详解

2018/10/18 Python

Python3.4学习笔记之常用操作符,条件分支和循环用法示例

2019/03/01 Python

基于python的docx模块处理word和WPS的docx格式文件方式

2020/02/13 Python

Python如何爬取qq音乐歌词到本地

2020/06/01 Python

html5 viewport使用方法示例详解

2013/12/02 HTML / CSS

集团公司总经理岗位职责

2013/12/20 职场文书

副乡长群众路线教育实践活动个人对照检查材料

2014/09/19 职场文书

群众路线专项整治方案

2014/10/27 职场文书

求职简历自我评价范文

2015/03/10 职场文书

小学五年级班主任工作经验交流材料

2015/11/02 职场文书

班主任远程培训研修日志

2015/11/13 职场文书

只需要12页，掌握撰写一流商业计划书的技巧

2019/05/07 职场文书

会计专业自荐信范文

2019/05/22 职场文书

北京大学中文系教授推荐的10本小说

2019/08/08 职场文书