PyTorch 隨機(jī)數(shù)生成占用 CPU 過高的解決方法
更新時間：2020年01月13日 09:40:52 作者：lucasgyshen
今天小編就為大家分享一篇PyTorch 隨機(jī)數(shù)生成占用 CPU 過高的解決方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
PyTorch 隨機(jī)數(shù)生成占用 CPU 過高的問題
今天在使用 pytorch 的過程中，發(fā)現(xiàn) CPU 占用率過高。經(jīng)過檢查，發(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 中生成過大的矩陣，而 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 過高的解決方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python+selenium點(diǎn)擊網(wǎng)頁上指定坐標(biāo)的實(shí)例
今天小編就為大家分享一篇Python+selenium點(diǎn)擊網(wǎng)頁上指定坐標(biāo)的實(shí)例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
Python+opencv+pyaudio實(shí)現(xiàn)帶聲音屏幕錄制
今天小編就為大家分享一篇Python+opencv+pyaudio實(shí)現(xiàn)帶聲音屏幕錄制，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
本文主要給大家介紹了yaml語法、yaml儲存數(shù)據(jù)，封裝類讀取yaml配置文件，以及yaml的用法和其原理，有需要的朋友可以參考下，希望可以有所幫助2021-08-08
Python深度學(xué)習(xí)之使用Pytorch搭建ShuffleNetv2
這篇文章主要介紹了Python深度學(xué)習(xí)之使用Pytorch搭建ShuffleNetv2,文中有非常詳細(xì)的圖文示例,對正在學(xué)習(xí)python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-05-05
Windows和Linux下使用Python訪問SqlServer的方法介紹
這篇文章主要介紹了Windows和Linux下使用Python訪問SqlServer的方法介紹,本文講解了Windows下配置Python訪問Sqlserver、Linux下配置Python訪問SqlServer等內(nèi)容,需要的朋友可以參考下2015-03-03
今天小編就為大家分享一篇pip命令無法使用的解決方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
NumPy提供了多種存取數(shù)組內(nèi)容的文件操作函數(shù)。保存數(shù)組數(shù)據(jù)的文件可以是二進(jìn)制格式或者文本格式。這篇文章主要介紹了python利用numpy存取文件,需要的朋友可以參考下2019-09-09
當(dāng)前使用的是anaconda的3.8版本，無法正常下載tensorflow包，需要構(gòu)建虛擬環(huán)境使用3.7及以下的解釋器才可以，如何解決這個問題呢，下面小編給大家?guī)砹巳绾卧趐ycharm中使用tensorflow，感興趣的朋友參考下吧2021-11-11
這篇文章主要介紹了pytorch Dropout過擬合的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
Python實(shí)現(xiàn)日期判斷和加減操作詳解
這篇文章主要介紹了如何利用Python實(shí)現(xiàn)日期的判斷，以及對日期的加減操作，文中的示例代碼對我們學(xué)習(xí)或工作有一定的價值，需要的可以參考一下2022-01-01