编程 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中time模块与datetime模块在使用中的不同之处

Nov 24 Python

详解Python中的Descriptor描述符类

Jun 14 Python

python中set()函数简介及实例解析

Jan 09 Python

python实现俄罗斯方块游戏

Mar 25 Python

使用python将图片格式转换为ico格式的示例

Oct 22 Python

详解Python3中ceil()函数用法

Feb 19 Python

Python+selenium点击网页上指定坐标的实例

Jul 05 Python

Django框架静态文件使用/中间件/禁用ip功能实例详解

Jul 22 Python

Python爬虫:将headers请求头字符串转为字典的方法

Aug 21 Python

简单了解python 生成器列表推导式生成器表达式

Aug 22 Python

python基础之爬虫入门

May 10 Python

详解Python中__new__方法的作用

Mar 31 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+MySQL5.0中文乱码解决方法

2006/11/20 PHP

浅谈web上存漏洞及原理分析、防范方法（安全文件上存方法）

2013/06/29 PHP

php实现的简易扫雷游戏实例

2015/07/09 PHP

php使用GD2绘制几何图形示例

2017/02/15 PHP

Thinkphp实现站点静态化的方法详解

2017/03/21 PHP

PHP实现的登录页面信息提示功能示例

2017/07/24 PHP

Prototype 学习工具函数学习($方法)

2009/07/12 Javascript

jQuery下的动画处理总结

2013/10/10 Javascript

基于JS实现省市联动效果代码分享

2016/06/06 Javascript

AngularJS实现使用路由切换视图的方法

2017/01/24 Javascript

Node连接mysql数据库方法介绍

2017/02/07 Javascript

基于JavaScript实现本地图片预览

2017/02/08 Javascript

浅谈Vue的加载顺序探讨

2017/10/25 Javascript

JavaScript中EventLoop介绍

2018/01/22 Javascript

vue实现树形菜单效果

2018/03/19 Javascript

vue文件树组件使用详解

2018/03/29 Javascript

JavaScript防止全局变量污染的方法总结

2018/08/02 Javascript

vue项目首屏加载时间优化实战

2019/04/23 Javascript

jQuery实现增删改查

2020/12/22 jQuery

[01:20]DOTA2上海特级锦标赛现场采访：谁的ID最受青睐

2016/03/25 DOTA

实例讲解Python中global语句下全局变量的值的修改

2016/06/16 Python

解决Linux系统中python matplotlib画图的中文显示问题

2017/06/15 Python

Python装饰器原理与简单用法实例分析

2018/04/29 Python

浅谈python中对于json写入txt文件的编码问题

2018/06/07 Python

[原创]Python入门教程1. 基本运算【四则运算、变量、math模块等】

2018/10/28 Python

selenium+python自动化测试之使用webdriver操作浏览器的方法

2019/01/23 Python

django2.2安装错误最全的解决方案(小结)

2019/09/24 Python

Python 中的pygame安装与配置教程详解

2020/02/10 Python

python新式类和经典类的区别实例分析

2020/03/23 Python

python 实现图像快速替换某种颜色

2020/06/04 Python

用React加CSS3实现微信拆红包动画效果

2017/03/13 HTML / CSS

自我鉴定总结

2014/03/24 职场文书

区域销售主管岗位职责

2014/06/15 职场文书

银行稽核岗位职责

2015/04/13 职场文书

再见，2019我们不负使命；你好，2020我们砥砺前行

2020/01/03 职场文书

利用python做表格数据处理

2021/04/13 Python