编程 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实现的简单dns查询功能示例

May 24 Python

python实现机械分词之逆向最大匹配算法代码示例

Dec 13 Python

Python中类的创建和实例化操作示例

Feb 27 Python

解决python中导入win32com.client出错的问题

Jul 26 Python

Python 动态变量名定义与调用方法

Feb 09 Python

Python3 集合set入门基础

Feb 10 Python

OpenCV中VideoCapture类的使用详解

Feb 14 Python

python 通过使用Yolact训练数据集

Apr 06 Python

Python开发之QT解决无边框界面拖动卡屏问题(附带源码)

May 27 Python

python_tkinter事件类型详情

Mar 20 Python

python开发制作好看的时钟效果

May 02 Python

移除Selenium中window.navigator.webdriver值

Jun 10 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

Cakephp 执行主要流程

2010/03/24 PHP

php获取百度收录、百度热词及百度快照的方法

2015/04/02 PHP

php无限级评论嵌套实现代码

2018/04/18 PHP

简单实用的PHP文本缓存类实例

2019/03/22 PHP

JS代码放在head和body中的区别分析

2011/12/01 Javascript

js时间日期和毫秒的相互转换

2013/02/22 Javascript

JS异常处理的一个想法(sofish)

2013/03/14 Javascript

使用Grunt.js管理你项目的应用说明

2013/04/24 Javascript

jquery实现控制表格行高亮实例

2013/06/05 Javascript

javascript学习笔记之10个原生技巧

2014/05/21 Javascript

12个超实用的JQuery代码片段

2015/11/02 Javascript

探讨JavaScript标签位置的存放与功能有无关系

2016/01/15 Javascript

JavaScript使用正则表达式获取全部分组内容的方法示例

2017/01/17 Javascript

浅谈$_FILES数组为空的原因

2017/02/16 Javascript

Vue如何引入远程JS文件

2017/04/20 Javascript

JavaScript实现瀑布流图片效果

2017/06/30 Javascript

JS实现图片旋转动画效果封装与使用示例

2018/07/09 Javascript

[01:20]DOTA2 2017国际邀请赛冠军之路无止竞

2017/06/19 DOTA

python实现在sqlite动态创建表的方法

2015/05/08 Python

Python3.5编程实现修改IIS WEB.CONFIG的方法示例

2017/08/18 Python

浅谈numpy数组的几种排序方式

2017/12/15 Python

Python 批量合并多个txt文件的实例讲解

2018/05/08 Python

python查看模块安装位置的方法

2018/10/16 Python

python使用xlrd模块读取xlsx文件中的ip方法

2019/01/11 Python

python 获得任意路径下的文件及其根目录的方法

2019/02/16 Python

Python supervisor强大的进程管理工具的使用

2019/04/24 Python

Python实现检测文件的MD5值来查找重复文件案例

2020/03/12 Python

Python sorted排序方法如何实现

2020/03/31 Python

拉夫劳伦爱尔兰官方网站：Ralph Lauren爱尔兰

2020/04/10 全球购物

c++工程师面试问题

2013/08/04 面试题

如何写求职信

2014/05/24 职场文书

花坛标语大全

2014/06/30 职场文书

基层党员群众路线教育实践活动个人对照检查材料思想汇报

2014/10/05 职场文书

2014年幼儿园园务工作总结

2014/12/05 职场文书

2015年春节标语口号

2014/12/09 职场文书

python可视化之颜色映射详解

2021/09/15 Python