PyTorch如何限制CPU線程數(shù)目
更新時(shí)間：2024年02月20日 10:30:48 作者：培之
這篇文章主要介紹了PyTorch如何限制CPU線程數(shù)目問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教
PyTorch限制CPU線程數(shù)目
終端命令 htop ：查看 CPU 利用率
限制 python 的cpu 利用率：
import os cpu_num = 4 # 這里設(shè)置成你想運(yùn)行的CPU個(gè)數(shù) os.environ["OMP_NUM_THREADS"] = str(cpu_num) # noqa os.environ["MKL_NUM_THREADS"] = str(cpu_num) # noqa torch.set_num_threads(cpu_num )
原因
tensor 沒(méi)有 .to('cuda') ,tensor 還是部署在 cpu 上，耗的是 cpu 的資源，導(dǎo)致 cpu 占用率過(guò)高。
在讀數(shù)據(jù)的時(shí)候設(shè)置 num_workers = 0，也可能會(huì)導(dǎo)致 cpu 占用率過(guò)高。
PyTorch程序占用較高CPU解決
在使用服務(wù)器運(yùn)行PyTorch程序時(shí)，使用TOP命令查看CPU狀態(tài)，發(fā)現(xiàn)CPU占用率竟然達(dá)到2000%，已經(jīng)影響到其他人正常使用服務(wù)器。
解決辦法
在main程序中加以下代碼：
torch.set_num_threads(1)
設(shè)置最多使用1個(gè)CPU核。問(wèn)題解決~
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python正則表達(dá)式匹配日期與時(shí)間的方法
這篇文章主要介紹了Python正則表達(dá)式匹配日期與時(shí)間的方法，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Python模塊的制作方法,結(jié)合實(shí)例形式分析了Python第三方模塊的編寫、調(diào)用相關(guān)操作技巧與使用注意事項(xiàng),需要的朋友可以參考下2019-12-12
Python3中的f-Strings增強(qiáng)版字符串格式化方法
這篇文章主要介紹了Python3中的f-Strings增強(qiáng)版字符串格式化方法,看完本文你將學(xué)習(xí)到如何以及為什么使用f-strings。對(duì)大家的工作或?qū)W習(xí)具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-03-03
Python的加密模塊md5、sha、crypt使用實(shí)例
這篇文章主要介紹了Python的加密模塊md5、sha、crypt使用實(shí)例,本文給出了MD5和crypt模塊的代碼實(shí)例,需要的朋友可以參考下2014-09-09
今天給大家?guī)?lái)的是關(guān)于Python的相關(guān)知識(shí),文章圍繞著Python print函數(shù)的用法展開,文中有非常詳細(xì)的介紹及代碼示例,需要的朋友可以參考下2021-06-06
這篇文章主要為大家詳細(xì)介紹了Python+OpenCV進(jìn)行讀寫視頻操作的示例代碼，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以動(dòng)手嘗試一下2022-08-08
python可視化分析繪制帶趨勢(shì)線的散點(diǎn)圖和邊緣直方圖
這篇文章主要介紹了python可視化分析繪制帶趨勢(shì)線的散點(diǎn)圖和邊緣直方圖，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，需要的小伙伴可以參考一下2022-06-06
python調(diào)用文字識(shí)別OCR輕松搞定驗(yàn)證碼
本文主要介紹了python調(diào)用文字識(shí)別OCR輕松搞定驗(yàn)證碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-11-11
如何使用pandas對(duì)超大csv文件進(jìn)行快速拆分詳解
Pandas是Python語(yǔ)言的一個(gè)擴(kuò)展程序庫(kù),提供高性能、易于使用的數(shù)據(jù)結(jié)構(gòu)和數(shù)據(jù)分析工具,下面這篇文章主要給大家介紹了關(guān)于如何使用pandas對(duì)超大csv文件進(jìn)行快速拆分的相關(guān)資料,需要的朋友可以參考下2022-07-07
詳解Python圖像形態(tài)學(xué)處理(開運(yùn)算,閉運(yùn)算,梯度運(yùn)算)
數(shù)學(xué)形態(tài)學(xué)（Mathematical Morphology）是一種應(yīng)用于圖像處理和模式識(shí)別領(lǐng)域的新方法。本文將為大家介紹Python圖像形態(tài)學(xué)處理中的開運(yùn)算、閉運(yùn)算和梯度運(yùn)算，感興趣的可以了解一下2022-06-06