PyTorch 隨機(jī)數(shù)生成占用 CPU 過(guò)高的解決方法
更新時(shí)間：2020年01月13日 09:40:52 作者：lucasgyshen
今天小編就為大家分享一篇PyTorch 隨機(jī)數(shù)生成占用 CPU 過(guò)高的解決方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
PyTorch 隨機(jī)數(shù)生成占用 CPU 過(guò)高的問(wèn)題
今天在使用 pytorch 的過(guò)程中，發(fā)現(xiàn) CPU 占用率過(guò)高。經(jīng)過(guò)檢查，發(fā)現(xiàn)是因?yàn)橄仍?CPU 中生成了隨機(jī)數(shù)，然后再調(diào)用.to(device)傳到 GPU，這樣導(dǎo)致效率變得很低，并且CPU 和 GPU 都被消耗。
查閱PyTorch文檔后發(fā)現(xiàn)，torch.randn(shape, out)可以直接在GPU中生成隨機(jī)數(shù)，只要shape是tensor.cuda.Tensor類型即可。這樣，就可以避免在 CPU 中生成過(guò)大的矩陣，而 shape 變量是很小的。
因此，下面的代碼就可以進(jìn)行這種操作了。
noise = torch.cuda.FloatTensor(shape) if torch.cuda.is_available() else torch.FloatTensor(shape) torch.randn(shape, out=noise)
以上這篇PyTorch 隨機(jī)數(shù)生成占用 CPU 過(guò)高的解決方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python中對(duì)字典的幾個(gè)處理方法分享，文章圍繞主題展開(kāi)詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，感興趣的小伙伴可以參考一下2022-08-08
python實(shí)現(xiàn)宿舍管理系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)宿舍管理系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-11-11
在本篇文章里小編給大家整理的是關(guān)于python實(shí)發(fā)郵件的相關(guān)知識(shí)點(diǎn)內(nèi)容，有需要的朋友們學(xué)習(xí)下。2019-11-11
python數(shù)據(jù)批量寫入ScrolledText的優(yōu)化方法
今天小編就為大家分享一篇python數(shù)據(jù)批量寫入ScrolledText的優(yōu)化方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-10-10
python獲取文件版本信息、公司名和產(chǎn)品名的方法
這篇文章主要介紹了python獲取文件版本信息、公司名和產(chǎn)品名的方法,是Python程序設(shè)計(jì)中非常實(shí)用的技巧,需要的朋友可以參考下2014-10-10
本文主要介紹了pandas按某列降序的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-01-01
python Yaml、Json、Dict之間的轉(zhuǎn)化
這篇文章主要介紹了python Yaml 、Json 、Dict 之間的轉(zhuǎn)化的示例，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下2020-10-10
這篇文章主要介紹了Python3 讀、寫Excel文件的操作方法,需要的朋友可以參考下2018-10-10
這篇文章主要介紹了python將下載到本地m3u8視頻合成MP4的代碼詳解,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-11-11
python格式化輸出保留2位小數(shù)的實(shí)現(xiàn)方法
這篇文章主要介紹了python格式化輸出保留2位小數(shù)的實(shí)現(xiàn)方法,需要的朋友可以參考下2019-07-07