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
投稿猜你喜欢
- length: 是计算字段的长度一个汉字是算三个字符,一个数字或字母算一个字符CHAR_LENGTH(str) 返回
Scrapy是一个用Python实现的为了爬取网站数据、提取数据的应用框架。我们对于爬取到的数据存储到本地或数据库是经常要用到的操作。主要讲- 本文用python实现线性回归算法，供大家参考，具体内容如下# -*- coding: utf-8 -*-"""
- Python 是一种代表简单思想的语言，其语法相对简单，很容易上手。不过，如果就此小视 Python 语法的精妙和深邃，那就大错特错了。本文
- 前言数据库索引是mysql数据库中重要的组成部分，是数据库查询数据速度提升的关键，本文将介绍数据库索引的一些内容。数据库索引的数据结构在数据
- 建造者模式建造者模式用于创建复杂的对象。使用建造者模式可以使复杂的过程层次明了、清晰，把对象的创建以及使用进行了解耦。实际上从代码的角度上看
- 建表，主键自增create table aaa(id bigint identity(1,1) not null PRIMARY key,n
- 不好意思，标题比较啰嗦，因为这次的流水账确实属于一个比较细节的小东西，下面详细讲：1需求最近在使用electron-vue 开发一个跨平台的
- 一、新建项目，在主配置文件中，修改以下内容：ALLOWED_HOSTS = ['127.0.0.1','localh
- 本文对于先前系列文章中实现的C/Python代码统计工具(CPLineCounter)，通过C扩展接口重写核心算法加以优化，并与网上常见的统
- 最近在学习Vue.js，把自己遇到的问题做个记录，所以，今天添加一点小笔记。在项目中遇到两个问题，简单的做个笔记来记录自己解决的问题，可能不
- 顾名思义, 条件查询就是使用where字句 , 将满足条件的数据筛选出来语法 :select < 结果 > from <
- 描述微软开发的两个动态库存在后门允许用户查看ASP文件源程序和下载整个网站详细随IIS和Frontpage Extention server
- 引用集：using System.Data.OleDb;static string exePath = System.Environment
- 引言django提供了一个默认的auth系统用于用户的登录和授权，并提供了一定的扩展性，允许开发者自行定义多个验证后台，每个验证后台必须实现
- 现在向大家介绍mysql命令行下，从数据库的建立到表数据的删除全过程，希望对大家有所帮助。登陆mysql打cmd命令终端，如果已经添加了my
- 本文实例讲述了Python实现的简单模板引擎功能。分享给大家供大家参考，具体如下：#coding:utf- 8__author__=&quo
- MySQL 提供了一个很有意思的Engine：Federated！如果你了解Linux下面的Link的话，就应该很好理解这个Federate
- 本文实例为大家分享了pygame实现雷电游戏开发代码，供大家参考，具体内容如下源代码:stars.py#-*- coding=utf-8 -
- 本文实例为大家分享了python实现学生信息管理系统的具体代码，供大家参考，具体内容如下学生管理系统的开发步骤：1、显示学生管理系统的功能菜
