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)
以上这篇PyTorch 随机数生成占用 CPU 过高的解决方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持小牛知识库。
-
本文向大家介绍Pytorch生成随机数Tensor的方法汇总，包括了Pytorch生成随机数Tensor的方法汇总的使用技巧和注意事项，需要的朋友参考一下 在使用PyTorch做实验时经常会用到生成随机数Tensor的方法，比如： torch.rand() torch.randn() torch.normal() torch.linespace() 均匀分布 torch.rand(*sizes,
-
本文向大家介绍java实战CPU占用过高问题的排查及解决，包括了java实战CPU占用过高问题的排查及解决的使用技巧和注意事项，需要的朋友参考一下 最近一段时间 某台服务器上的一个应用总是隔一段时间就自己挂掉 用top看了看 从重新部署应用开始没有多长时间CPU占用上升得很快 排查步骤 1.使用top 定位到占用CPU高的进程PID top 2.通过ps aux | grep PID命令 获取线程
-
使用的是 2 核的腾讯云轻量服务器,这个 zapppp 进程几乎占用了所有 CPU 资源,有可能是一个挖矿程序,使用常规的方法都无法终结这个进程。前几天收到邮件说服务器的 22 端口存在攻击行为,我怀疑是攻击者通过破解 ssh 登录密码登录到服务器然后植入恶意程序。 请问一下如何完整地修复这个问题?
-
今天，我发现我的服务器的cpu负载过高，而服务器只是在运行一个Java应用程序。 下面是我的操作步骤。 > 我使用命令查找应用程序的PID。pid为25713。 以下是我的问题： 为什么使cpu负载过高。 为什么在我使用命令后，cpu变得正常。 不止这一次，每一次。 当我执行命令时，打印的日志
-
本文向大家介绍Python随机生成手机号、数字的方法详解，包括了Python随机生成手机号、数字的方法详解的使用技巧和注意事项，需要的朋友参考一下 本文实例讲述了Python随机生成手机号、数字的方法。分享给大家供大家参考，具体如下： Python随机产生手机号、数字。代码如下： 函数简介： （1）randrange：这个函数会产生范围是4~9之间的任意一个数字，注意不包含10. （2）choic
-
random 生成随机数包 文档：https://www.npmjs.com/package/random 安装：npm install --save random 封装代码： app / extend / context.js // 导入 jwt const jwt = require('jsonwebtoken') // 导入随机数包 const random = require('rando