PyTorch 随机数生成占用 CPU 过高的解决方法
作者：lucasgyshen 发布时间：2021-09-23 09:12:31
标签：PyTorch,随机数,生成,CPU
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)
来源：https://blog.csdn.net/a463560470/article/details/82985905
0
投稿猜你喜欢
- HTML5 中 div section article 的区别刚刚开始接触 HTML5 时，对它的标签很不适应，甚至一度有点反感。尤其是对
- 首先，FSO是FileSystemObject的简称。当然也就是我们的俗称FSO组件了，该组件可以用来处理驱动器、文件夹以及文件。它可以检测
- 概述很多人接触Python，都是从爬虫开始，其实很多语言都可以做爬虫，只是Python相对其他语言来说，更加简单而已。但是Python并不止
目前由于phantomjs已经不维护了，而新版的Chrome（59+）推出了Headless模式，对爬虫来说尤其是定时任务的爬虫截屏之类的是- 你一定想下载一下感兴趣的网页，以便慢慢欣赏吧！利用FrontPage能够轻松做到这一点，甚至可以下载整个站点，当然这里只能下载静态的页面。启
- 用过QQ的人应该都知道软键盘，他可以增强我们密码的安全性，以保证我们的密码资料不被非法监听。现在软键盘也用在了很多的网站上，像早期银行的在线
- 这是我研究网页切片算法的一个汇总想法。之前我写过:一种面向搜索引擎的网页分块、切片的原理，实现和演示，随着工作的深入，逐渐碰到以
- 今天是Firefox3的2008下载日(貌似北京时间是6.18的凌晨1:00)，这就意味着Firefox3正式发布了。Firefox3有众多
- 本文实例讲述了Thinkphp5.0 框架的请求方式与响应方式。分享给大家供大家参考，具体如下：Thinkphp5.0 的请求方式方法一（使
- 本文实例讲述了python中enumerate函数遍历元素用法。分享给大家供大家参考，具体如下：enumerate函数用于遍历序列中的元素以
- 1**：请求收到，继续处理2**：操作成功收到，分析、接受3**：完成此请求必须进一步处理4**：请求包含一个错误语法或不能完成5**：服务
- 相关推荐：完整的sql中文参考手册（chm）下载 DB2 提供了关连式资料库的查询语言 sql (Structured Query
- PHP mysqli_select_db() 函数更改连接的默认数据库：删除数据库<?php // 假定数据库用户名：root，密码：
- 一、Pytorch创建张量的4种方法Pytorch创建张量的4种方法主要有：torch.Tensor()、torch.tensor()、to
- 缘起最近复习设计模式拜读谭勇德的<<设计模式就该这样学>>该书以java语言演绎了常见设计模式本系列笔记拟采用gol
- 在我们建立一个数据库时，并且想将分散在各处的不同类型的数据库分类汇总在这个新建的数据库中时，尤其是在进行数据检验、净化和转换时，将会面临很大
- virtualenvwrapper是用来管理virtualenv的扩展包，用着很方便。1. 安装：#安装virtualenvwrapper$
- sys模块：全称system，指的是解释器。常用操作，用于接收系统操作系统调用解释器传入的参数1、 sys.argv获取脚本传递的所有参数，
- 如下所示：区别ArrayTensor类型uint8，float32系列{}各类型相互转换uint8转float64：image = imag
- 这篇文章主要介绍了Python如何实现强制数据类型转换,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
